Función torch.concat de PyTorch

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

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

Definición de la función

torch.concat(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): el tensor de salida.

Valor de retorno:

  • torch.Tensor: devuelve el tensor concatenado.

Ejemplos de uso

Ejemplo

import torch

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

# Concatenar dos tensores
result = torch.concat([a, b])

print(result)

El resultado es:

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

Ejemplo

import torch

# Crear dos tensores bidimensionales
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])

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

# Concatenar a lo largo de la segunda dimensión
d = torch.concat([a, b], dim=1)

print("Concatenar a lo largo de dim=0:")
print(c)
print("nConcatenar a lo largo de dim=1:")
print(d)

El resultado es:

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

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

Nota:torch.concatSítorch.catel alias de, las funciones de ambos son exactamente iguales.

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

Otras extensiones