Fonction torch.permute de PyTorch
Manuel de référence Pytorch torch
torch.permuteest une fonction de PyTorch utilisée pour réorganiser les dimensions d'un tenseur.
Définition de fonction
torch.permute(input, dims)
Exemples d'utilisation
Exemple
import torch
# Créer un tenseur 3D
x = torch.randn(2, 3, 4)
print("Forme d'origine :", x.shape)
# Réorganiser les dimensions (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("Après permute :", y.shape)
# Créer un tenseur 3D
x = torch.randn(2, 3, 4)
print("Forme d'origine :", x.shape)
# Réorganiser les dimensions (2,3,4) -> (4,2,3)
y = x.permute(2, 0, 1)
print("Après permute :", y.shape)
Autres extensions