Tenseur Identité en PyTorch

C’est l’élément neutre de la muliplication matricielle. L’une des matrices les plus fondamentales et les plus simples : la matrice identité.

Lorsque l’on débute avec PyTorch, on ne peut pas deviner comment instancier facilement une matrice identité.

Alors il ne faut pas avoir honte de demander comment l’on peut créer une matrice, ou en l’occurrence un tenseur, qui représente l’identité avec PyTorch.

🚩Problème :

Comment créer l’identité en PyTorch?

Solution :

Utiliser la fonction eye de torch.

Exemple :

import torch
identity = torch.eye(n) # avec n la dimension de la matrice identité.

Laisser un commentaire