Función torch.concatenate de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.concatenatees una función en PyTorch utilizada para concatenar múltiples tensores a lo largo de una dimensión especificada. Estorch.cates la misma función, utilizada para unir varios tensores a lo largo de una dimensión especificada en un tensor más grande.

Definición de la función

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

Parámetros:

  • tensors(Sequence of Tensor): la secuencia de tensores a concatenar. Todos los tensores deben tener la misma forma en todas las dimensiones excepto en la dimensión de concatenación.
  • dim(int, opcional): la dimensión de concatenación, por defecto 0.
  • out(Tensor, opcional): tensor de salida.

Valor de retorno:

  • torch.Tensor: Devuelve el tensor concatenado.

Ejemplo de uso

Ejemplo

import torch

# Crear múltiples tensores
a = torch.tensor([1, 2])
b = torch.tensor([3, 4])
c = torch.tensor([5, 6])

# Concatenar múltiples tensores
result = torch.concatenate([a, b, c])

print(result)

El resultado de salida es:

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

Ejemplo

import torch

# Crear dos tensores bidimensionales
a = torch.randn(2, 3)
b = torch.randn(2, 3)

# Concatenar a lo largo de la primera dimensión
c = torch.concatenate([a, b], dim=0)

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

El resultado de salida es:

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

Nota:torch.concatenateSítorch.cates un alias de, las funciones de ambos son completamente idénticas. En el código real, el más utilizado estorch.cat。

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones