Fonction torch.atleast_1d de PyTorch
Manuel de référence PyTorch torch
torch.atleast_1dest une fonction dans PyTorch utilisée pour convertir les tenseurs d'entrée en au moins 1 dimension. Si l'entrée est déjà au moins unidimensionnelle, elle renvoie une vue du tenseur d'origine ; sinon, elle ajoute une nouvelle dimension.
Définition de fonction
torch.atleast_1d(*tensors)
Exemple d'utilisation
Exemple
import torch
# Scalaire converti en tenseur 1-D
x = torch.atleast_1d(5)
print("Après conversion du scalaire :", x, "Forme :", x.shape)
# Tenseur déjà 1-D
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("tenseur 1-D :", y, "Forme :", y.shape)
# Le tenseur 2-D reste inchangé
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("tenseur 2-D :", y, "Forme :", y.shape)
# Entrées multiples
a = 1
b = torch.tensor([2, 3])
c = torch.tensor([[4, 5]])
result = torch.atleast_1d(a, b, c)
for i, t in enumerate(result):
print(f"Entrée {i} forme : {t.shape}")
# Scalaire converti en tenseur 1-D
x = torch.atleast_1d(5)
print("Après conversion du scalaire :", x, "Forme :", x.shape)
# Tenseur déjà 1-D
x = torch.tensor([1, 2, 3])
y = torch.atleast_1d(x)
print("tenseur 1-D :", y, "Forme :", y.shape)
# Le tenseur 2-D reste inchangé
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_1d(x)
print("tenseur 2-D :", y, "Forme :", y.shape)
# Entrées multiples
a = 1
b = torch.tensor([2, 3])
c = torch.tensor([[4, 5]])
result = torch.atleast_1d(a, b, c)
for i, t in enumerate(result):
print(f"Entrée {i} forme : {t.shape}")
Autres extensions