PyTorch torch.atleast_2d Funktion
PyTorch torch Referenzhandbuch
torch.atleast_2dist eine Funktion in PyTorch, die verwendet wird, um Eingabetensoren in mindestens 2 Dimensionen umzuwandeln. Wenn die Eingabe weniger als 2 Dimensionen hat, fügt sie automatisch Dimensionen hinzu, um sie 2-dimensional zu machen.
Funktionsdefinition
torch.atleast_2d(*tensors)
Anwendungsbeispiel
]) Form: torch.Size([1, 1])
import torch
# Skalar in einen 2-dimensionalen Tensor umwandeln
x = torch.atleast_2d(5)
print("Nach Skalarkonvertierung:", x, "Form:", x.shape)
# Ausgabe: Nach der Skalarkonvertierung: tensor([
# 1-dimensionalen Tensor in 2 Dimensionen umwandeln
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1D zu 2D:", y, "Form:", y.shape)
# Ausgabe: 1D zu 2D: tensor([[1, 2, 3]]) Form: torch.Size([1, 3])
# Ein Tensor, der bereits 2-dimensional ist, bleibt unverändert
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2D-Tensor:", y, "Form:", y.shape)
# Skalar in einen 2-dimensionalen Tensor umwandeln
x = torch.atleast_2d(5)
print("Nach Skalarkonvertierung:", x, "Form:", x.shape)
# Ausgabe: Nach der Skalarkonvertierung: tensor([
# 1-dimensionalen Tensor in 2 Dimensionen umwandeln
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1D zu 2D:", y, "Form:", y.shape)
# Ausgabe: 1D zu 2D: tensor([[1, 2, 3]]) Form: torch.Size([1, 3])
# Ein Tensor, der bereits 2-dimensional ist, bleibt unverändert
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("2D-Tensor:", y, "Form:", y.shape)
Weitere Erweiterungen