Función torch.vstack de PyTorch

Manual de referencia de Pytorch torch

torch.vstackEs una función en PyTorch que se utiliza para apilar tensores verticalmente (a lo largo de las filas).

Definición de funciones

torch.vstack(tensors, *, out=None)

Ejemplo de uso

Ejemplos

import torch

# Apilamiento vertical de tensores unidimensionales
x1 = torch.tensor([1, 2, 3])
x2 = torch.tensor([4, 5, 6])
result = torch.vstack([x1, x2])
print("Apilado vertical de tensores unidimensionales:")
print(f"  x1: {x1}")
print(f"  x2: {x2}")
print(f"  vstack:n{result}")

# Apilamiento vertical de tensores bidimensionales
y1 = torch.tensor([[1, 2], [3, 4]])
y2 = torch.tensor([[5, 6], [7, 8]])
result = torch.vstack([y1, y2])
print("nApilamiento vertical de tensores 2D:")
print(f"  y1:n{y1}")
print(f"  y2:n{y2}")
print(f"  vstack:n{result}")

# Apilar múltiples tensores
z1 = torch.tensor([1, 2, 3])
z2 = torch.tensor([4, 5, 6])
z3 = torch.tensor([7, 8, 9])
result = torch.vstack([z1, z2, z3])
print("nApilamiento de múltiples tensores 1D:")
print(f"  result:n{result}")

El resultado de salida es:

一维张量垂直堆叠:
  x1: tensor([1, 2, 3])
  x2: tensor([4, 5, 6])
  vstack:
tensor([[1, 2, 3],
        [4, 5, 6]])

二维张量垂直堆叠:
  y1:
tensor([[1, 2],
        [3, 4]])
  y2:
tensor([[5, 6],
        [7, 8]])
  vstack:
tensor([[1, 2],
        [3, 4],
        [5, 6],
        [7, 8]])

多个一维张量堆叠:
  result:
tensor([[1, 2, 3],
        [4, 5, 6],
        [7, 8, 9]])

Manual de referencia de Pytorch torch

Otras extensiones