Fonction torch.squeeze de PyTorch

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

torch.squeezeC'est une fonction de PyTorch utilisée pour supprimer les dimensions de taille 1 d'un tenseur.

Définition de fonction

torch.squeeze(input, dim)

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur avec des dimensions de taille 1
x = torch.randn(1, 2, 1, 3, 1)
print("Forme d'origine :", x.shape)

# Supprimer toutes les dimensions de taille 1
y = torch.squeeze(x)
print("Forme après squeeze :", y.shape)

# Spécifier la dimension
z = torch.squeeze(x, dim=0)
print("Après squeeze dim=0 :", z.shape)

La sortie est :

原始形状: torch.Size([1, 2, 1, 3, 1])
挤压后形状: torch.Size([2, 3])
dim=0 挤压后: torch.Size([2, 1, 3, 1])

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

Autres extensions