Fonction torch.diag de PyTorch

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

torch.diagEst une fonction de PyTorch utilisée pour créer des matrices diagonales ou extraire les éléments diagonaux d’un tenseur.

Définition de fonction

torch.diag(input, diagonal=0, out=None)

Description des paramètres :

  • input: tenseur d’entrée
  • diagonal: indice de diagonale, 0 représente la diagonale principale, les valeurs positives représentent les diagonales supérieures, les valeurs négatives représentent les diagonales inférieures

Exemple d'utilisation

Exemple

import torch

# Créer un tenseur unidimensionnel
x = torch.tensor([1, 2, 3])

# Créer une matrice diagonale
y = torch.diag(x)
print(y)

Le résultat de sortie est:

tensor([[1, 0, 0],
        [0, 2, 0],
        [0, 0, 3]])

Exemple

import torch

# Extraire les éléments diagonaux d’une matrice
x = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Extraire la diagonale principale
y = torch.diag(x)
print(y)

Le résultat de sortie est:

tensor([1, 5, 9])

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

Autres extensions