Функция torch.concatenate в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.concatenateявляется функцией в PyTorch, используемой для конкатенации нескольких тензоров вдоль указанного измерения. Она совпадает сtorch.catявляется той же функцией, используемой для объединения нескольких тензоров вдоль указанного измерения в более крупный тензор.

Определение функции

torch.concatenate(tensors, dim=0, out=None)

Parameters:

  • tensors(Последовательность Tensor): Последовательность тензоров для конкатенации. Все тензоры должны иметь одинаковую форму во всех измерениях, кроме измерения конкатенации.
  • dim(int, optional): Измерение, по которому выполняется конкатенация; по умолчанию 0.
  • out(Tensor, необязательный): Выходной тензор.

Возвращаемое значение:

  • torch.Tensor: Возвращает конкатенированный тензор.

Примеры использования

Пример

import torch

# Создать несколько тензоров
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])
c = torch.tensor([5, 6])

# Конкатенация нескольких тензоров
result = torch.concatenate([a, b, c])

print(result)

Вывод:

tensor([1, 2, 3, 4, 5, 6])

Пример

import torch

# Создайте два 2D тензора
a = torch.randn(2, 3)
b = torch.randn(2, 3)

# Конкатенация по первому измерению
c = torch.concatenate([a, b], dim=0)

print("Размер a:", a.shape)
print("Размер b:", b.shape)
print("Размер c:", c.shape)

Вывод:

a 的形状: torch.Size([2, 3])
b 的形状: torch.Size([2, 3])
c 的形状: torch.Size([4, 3])

Примечание:torch.concatenateДаtorch.catэто псевдоним, и оба имеют абсолютно одинаковую функциональность. В реальном коде чаще используетсяtorch.cat。

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения