Fonction PyTorch torch.concat

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

torch.concat是 PyTorch 中的一个函数，用于沿指定维度拼接多个张量。它torch.cat是同一个函数，用于沿指定维度将多个张量拼接成一个更大的张量。

Définition de fonction

torch.concat(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，可选)：拼接的维度，默认是 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 deux tenseurs
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])

# 拼接两个张量
result = torch.concat([a, b])

print(result)

Le résultat de sortie est:

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

Exemple

import torch

# Créer deux tenseurs 2D
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

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

# Concaténer selon la deuxième dimension
d = torch.concat([a, b], dim=1)

print("Concaténation selon dim=0:")
print(c)
print("nConcaténation selon dim=1:")
print(d)

Le résultat de sortie est:

沿 dim=0 连接:
tensor([[1, 2],
        [3, 4],
        [5, 6],
        [7, 8]])

沿 dim=1 连接:
tensor([[1, 2, 5, 6],
        [3, 4, 7, 8]])

Remarque :torch.concatOuitorch.catest un alias, et la fonctionnalité des deux est exactement la même.

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

Autres extensions