Fonction torch.ravel de PyTorch
Manuel de référence torch de Pytorch
torch.ravelC'est une fonction de PyTorch utilisée pour aplatir un tenseur en une dimension. Elle renvoie une vue du tenseur d'entrée (si possible), pas une copie.
Définition de fonction
torch.ravel(input)
Description du paramètre :
input: tenseur d'entrée
Exemple d'utilisation
Exemple
import torch
# Créer une matrice 2x3
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Aplatir en une dimension
y = torch.ravel(x)
print(y)
# Créer une matrice 2x3
x = torch.tensor([[1, 2, 3], [4, 5, 6]])
# Aplatir en une dimension
y = torch.ravel(x)
print(y)
Le résultat de sortie est:
tensor([1, 2, 3, 4, 5, 6])
Exemple
import torch
# Créer un tenseur tridimensionnel
x = torch.arange(12).reshape(2, 3, 2)
# Aplatir en une dimension
y = torch.ravel(x)
print(y)
# Créer un tenseur tridimensionnel
x = torch.arange(12).reshape(2, 3, 2)
# Aplatir en une dimension
y = torch.ravel(x)
print(y)
Le résultat de sortie est:
tensor([ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11])
Autres extensions