Función torch.hstack de PyTorch

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

torch.hstackEs una función en PyTorch para apilar tensores horizontalmente (a lo largo de las columnas).

Definición de la función

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

Ejemplo de uso

Ejemplo

import torch

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

# Apilado horizontal de tensores bidimensionales
y1 = torch.tensor([[1, 2], [3, 4]])
y2 = torch.tensor([[5, 6], [7, 8]])
result = torch.hstack([y1, y2])
print("nApilado horizontal de tensores bidimensionales:")
print(f"  y1:n{y1}")
print(f"  y2:n{y2}")
print(f"  hstack:n{result}")

# Apilado de múltiples tensores
z1 = torch.tensor([1, 2])
z2 = torch.tensor([3, 4])
z3 = torch.tensor([5, 6])
result = torch.hstack([z1, z2, z3])
print("nApilado de múltiples tensores unidimensionales:")
print(f"  result: {result}")

El resultado de salida es:

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

二维张量水平堆叠:
  y1:
tensor([[1, 2],
        [3, 4]])
  y2:
tensor([[5, 6],
        [7, 8]])
  hstack:
tensor([[1, 2, 5, 6],
        [3, 4, 7, 8]])

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

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

Otras extensiones