PyTorch torch.concatenate 函数

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.concatenateest une fonction de PyTorch utilisée pour concaténer plusieurs tenseurs le long d'une dimension spécifiée. Elle est identique àtorch.cat是同一个函数，用于沿指定维度将多个张量拼接成一个更大的张量。

Définition de fonction

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

Parameters:

  • tensors(Séquence de Tensor) : La séquence de tenseurs à concaténer. Tous les tenseurs doivent avoir la même forme dans toutes les dimensions sauf la dimension de concaténation.
  • dim(int, facultatif) : La dimension selon laquelle concaténer, par défaut 0.
  • out(Tensor, optionnel) : Le tenseur de sortie.

Valeur de retour:

  • torch.Tensor: Renvoie le tenseur concaténé.

Exemples d'utilisation

Exemple

import torch

# Créer plusieurs tenseurs
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])
c = torch.tensor([5, 6])

# Concaténer plusieurs tenseurs
result = torch.concatenate([a, b, c])

print(result)

La sortie est :

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

Exemple

import torch

# Créer deux tenseurs 2D
a = torch.randn(2, 3)
b = torch.randn(2, 3)

# Concaténer selon la première dimension
c = torch.concatenate([a, b], dim=0)

print("Forme de a :", a.shape)
print("Forme de b :", b.shape)
print("Forme de c :", c.shape)

La sortie est :

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

Remarque :torch.concatenateOuitorch.catun alias, et les deux ont exactement la même fonctionnalité. Dans le code réel, le plus couramment utilisé esttorch.cat。

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions