129 votes

Que fait "unsqueeze" dans Pytorch?

Je n'arrive pas à comprendre comment l'exemple de la documentation PyTorch correspond à l'explication :

Renvoie un nouveau tenseur avec une dimension de taille un inséré à la position spécifiée. [...]

 >>> x = torch.tensor([1, 2, 3, 4])
>>> torch.unsqueeze(x, 0)
tensor([[ 1,  2,  3,  4]])
>>> torch.unsqueeze(x, 1)
tensor([[ 1],
        [ 2],
        [ 3],
        [ 4]])

Prograide.com

Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.

Powered by:

X