Que ce soit pour calculer une variance, une erreur quadratique moyenne, ou d’autres choses encore, calculer le carré d’un tenseur PyTorch est une tâche que vous pouvez être fréquemment amené à faire.
Dans certains cas, écrire simplement le produit du tenseur par lui-même fera l’affaire.
En revanche, dans le cas où le nom de la variable du tenseur est longue, on pourra utiliser avantageusement une fonction fournie avec PyTorch.
En termes de temps d’exécution, les deux approches se valent.
Problème :
Comment mettre au carré les éléments d’un tenseur en PyTorch?
Solution :
Utiliser la fonction square de torch.
Exemple :
La Data Science 
Laisser un commentaire