Fonction torch.atleast_2d de PyTorch

Pytorch torch 参考手册Manuel de référence PyTorch torch

torch.atleast_2dest une fonction de PyTorch utilisée pour convertir les tenseurs d'entrée en au moins 2 dimensions. Si l'entrée a moins de 2 dimensions, elle ajoute automatiquement des dimensions pour la rendre 2-dimensionnelle.

Définition de fonction

torch.atleast_2d(*tensors)

Exemple d'utilisation

]) Forme : torch.Size([1, 1])

import torch

# Convertir un scalaire en tenseur à 2 dimensions
x = torch.atleast_2d(5)
print("Après conversion du scalaire :", x, "Forme :", x.shape)
# Sortie : Après conversion du scalaire : tensor([

# Convertir un tenseur 1-dimensionnel en 2 dimensions
x = torch.tensor([1, 2, 3])
y = torch.atleast_2d(x)
print("1D vers 2D :", y, "Forme :", y.shape)
# Sortie : 1D vers 2D : tensor([[1, 2, 3]]) Forme : torch.Size([1, 3])

# Un tenseur déjà 2-dimensionnel reste inchangé
x = torch.tensor([[1, 2], [3, 4]])
y = torch.atleast_2d(x)
print("tenseur 2D :", y, "Forme :", y.shape)

Pytorch torch 参考手册Manuel de référence PyTorch torch

Autres extensions