PyTorch torch.ravel-Funktion
Pytorch torch-Referenzhandbuch
torch.ravelEs ist eine Funktion in PyTorch, mit der ein Tensor auf eine Dimension abgeflacht wird. Sie gibt eine Ansicht des Eingabetensors zurück (falls möglich), keine Kopie.
Funktionsdefinition
torch.ravel(input)
Parameter description:
input: Eingabetensor
Anwendungsbeispiel
Beispiel
import torch
# Erstelle eine 2x3-Matrix
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Auf eine Dimension abflachen
y = torch.ravel(x)
print(y)
# Erstelle eine 2x3-Matrix
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Auf eine Dimension abflachen
y = torch.ravel(x)
print(y)
Das Ausgabeergebnis ist:
tensor([1, 2, 3, 4, 5, 6])
Beispiel
import torch
# Erstelle einen dreidimensionalen Tensor
x = torch.arange(12).reshape(2, 3, 2)
# Auf eine Dimension abflachen
y = torch.ravel(x)
print(y)
# Erstelle einen dreidimensionalen Tensor
x = torch.arange(12).reshape(2, 3, 2)
# Auf eine Dimension abflachen
y = torch.ravel(x)
print(y)
Das Ausgabeergebnis ist:
tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
Weitere Erweiterungen