Fonction torch.hstack de PyTorch

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

torch.hstackC'est une fonction de PyTorch utilisée pour empiler des tenseurs horizontalement (le long des colonnes).

Définition de fonction

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

Exemple d'utilisation

Exemple

import torch

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

# Empilement horizontal de tenseurs bidimensionnels
y1 = torch.tensor([[1, 2], [3, 4]])
y2 = torch.tensor([[5, 6], [7, 8]])
result = torch.hstack([y1, y2])
print("nEmpilement horizontal de tenseurs 2D :")
print(f"  y1:n{y1}")
print(f"  y2:n{y2}")
print(f" hstack:n{result}")

# Empiler plusieurs tenseurs
z1 = torch.tensor([1, 2])
z2 = torch.tensor([3, 4])
z3 = torch.tensor([5, 6])
result = torch.hstack([z1, z2, z3])
print("nEmpiler plusieurs tenseurs 1D :")
print(f" résultat : {result}")

Le résultat de sortie est:

一维张量堆叠:
  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 参考手册Manuel de référence Pytorch torch

Autres extensions