Fonction torch.eye de PyTorch

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

torch.eyeest une fonction dans PyTorch utilisée pour créer une matrice identité (une matrice avec des 1 sur la diagonale et des 0 ailleurs).

Ceci est couramment utilisé en apprentissage profond pour l'initialisation, la création de masques et d'autres scénarios.

Définition de fonction

torch.eye(n, m, dtype, device, requires_grad)

Parameters:

  • n(int) : nombre de lignes.
  • m(int, facultatif) : nombre de colonnes. Si non spécifié, une matrice carrée n×n est créée.
  • dtype(torch.dtype, facultatif) : type de données.
  • device(torch.device, facultatif) : périphérique.
  • requires_grad(bool, facultatif) : si le calcul du gradient est nécessaire.

Valeur de retour:

  • torch.Tensor: renvoie la matrice identité.

Exemples d'utilisation

Exemple 1 : Créer une matrice carrée

Exemple

import torch

# Créer une matrice identité 3x3
I = torch.eye(3)

print(I)

La sortie est :

tensor([[1., 0., 0.],
        [0., 1., 0.],
        [0., 0., 1.]])

Exemple 2 : Créer une matrice non carrée

Exemple

import torch

# Créer une matrice identité 3x4
I = torch.eye(3, 4)

print(I)

La sortie est :

tensor([[1., 0., 0., 0.],
        [0., 1., 0., 0.],
        [0., 0., 1., 0.]])

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

Autres extensions