PyTorch torch.concat Funktion
PyTorch torch Referenzhandbuch
torch.concatist eine Funktion in PyTorch, mit der mehrere Tensoren entlang einer angegebenen Dimension verkettet werden. Sietorch.catist dieselbe Funktion, die verwendet wird, um mehrere Tensoren entlang einer angegebenen Dimension zu einem größeren Tensor zu verketten.
Funktionsdefinition
torch.concat(tensors, dim=0, out=None)
Parameters:
tensors(Sequenz von Tensor): Die Sequenz der zu verkettenden Tensoren. Alle Tensoren müssen in allen Dimensionen außer der Verkettungsdimension die gleiche Form haben.dim(int, optional): Die Dimension, entlang der verkettet wird, Standardwert ist 0.out(Tensor, optional): Der Ausgabe-Tensor.
Rückgabewert:
torch.Tensor: Gibt den verketteten Tensor zurück.
Anwendungsbeispiele
Beispiel
import torch
# Zwei Tensoren erstellen
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Zwei Tensoren verketten
result = torch.concat([a, b])
print(result)
# Zwei Tensoren erstellen
a = torch.tensor([1, 2, 3])
b = torch.tensor([4, 5, 6])
# Zwei Tensoren verketten
result = torch.concat([a, b])
print(result)
Das Ausgabeergebnis ist:
tensor([1, 2, 3, 4, 5, 6])
Beispiel
import torch
# Erstelle zwei 2D-Tensoren
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])
# Entlang der ersten Dimension verketten
c = torch.concat([a, b], dim=0)
# Entlang der zweiten Dimension verketten
d = torch.concat([a, b], dim=1)
print("Verkettung entlang dim=0:")
print(c)
print("nVerkettung entlang dim=1:")
print(d)
# Erstelle zwei 2D-Tensoren
a = torch.tensor([[1, 2], [3, 4]])
b = torch.tensor([[5, 6], [7, 8]])
# Entlang der ersten Dimension verketten
c = torch.concat([a, b], dim=0)
# Entlang der zweiten Dimension verketten
d = torch.concat([a, b], dim=1)
print("Verkettung entlang dim=0:")
print(c)
print("nVerkettung entlang dim=1:")
print(d)
Das Ausgabeergebnis ist:
沿 dim=0 连接:
tensor([[1, 2],
[3, 4],
[5, 6],
[7, 8]])
沿 dim=1 连接:
tensor([[1, 2, 5, 6],
[3, 4, 7, 8]])
Hinweis:torch.concatJatorch.catist ein Alias, und die Funktionalität der beiden ist genau dieselbe.
Andere Erweiterungen