PyTorch torch.hstack Funktion
PyTorch torch Referenzhandbuch
torch.hstackEs ist eine Funktion in PyTorch, mit der Tensoren horizontal (entlang der Spalten) gestapelt werden.
Funktionsdefinition
torch.hstack(tensors, *, out=None)
Anwendungsbeispiel
Beispiel
import torch
# Horizontales Stapeln eindimensionaler Tensoren
x1 = torch.tensor([1, 2, 3])
x2 = torch.tensor([4, 5, 6])
result = torch.hstack([x1, x2])
print("Stapeln von 1D-Tensoren:")
print(f" x1: {x1}")
print(f" x2: {x2}")
print(f" hstack: {result}")
# Horizontales Stapeln zweidimensionaler Tensoren
y1 = torch.tensor([[1, 2], [3, 4]])
y2 = torch.tensor([[5, 6], [7, 8]])
result = torch.hstack([y1, y2])
print("nHorizontales Stapeln von 2D-Tensoren:")
print(f" y1:n{y1}")
print(f" y2:n{y2}")
print(f" hstack:n{result}")
# Mehrere Tensoren stapeln
z1 = torch.tensor([1, 2])
z2 = torch.tensor([3, 4])
z3 = torch.tensor([5, 6])
result = torch.hstack([z1, z2, z3])
print("nStapeln mehrerer 1D-Tensoren:")
print(f" Ergebnis: {result}")
# Horizontales Stapeln eindimensionaler Tensoren
x1 = torch.tensor([1, 2, 3])
x2 = torch.tensor([4, 5, 6])
result = torch.hstack([x1, x2])
print("Stapeln von 1D-Tensoren:")
print(f" x1: {x1}")
print(f" x2: {x2}")
print(f" hstack: {result}")
# Horizontales Stapeln zweidimensionaler Tensoren
y1 = torch.tensor([[1, 2], [3, 4]])
y2 = torch.tensor([[5, 6], [7, 8]])
result = torch.hstack([y1, y2])
print("nHorizontales Stapeln von 2D-Tensoren:")
print(f" y1:n{y1}")
print(f" y2:n{y2}")
print(f" hstack:n{result}")
# Mehrere Tensoren stapeln
z1 = torch.tensor([1, 2])
z2 = torch.tensor([3, 4])
z3 = torch.tensor([5, 6])
result = torch.hstack([z1, z2, z3])
print("nStapeln mehrerer 1D-Tensoren:")
print(f" Ergebnis: {result}")
Das Ausgabeergebnis ist:
一维张量堆叠:
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])
Andere Erweiterungen