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]])