Fonction torch.diagflat de PyTorch

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

torch.diagflatIl s'agit d'une fonction de PyTorch utilisée pour créer une matrice diagonale plate. Que l'entrée soit unidimensionnelle ou multidimensionnelle, elle sera aplatie et utilisée comme éléments diagonaux.

Définition de fonction

torch.diagflat(input, diagonal=0)

Description des paramètres :

  • input: tenseur d'entrée, sera aplati
  • diagonal: indice diagonal

Exemple d'utilisation

Exemple

import torch

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

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

Le résultat de sortie est:

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

Exemple

import torch

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

# Sera aplati en [1,2,3,4], créant une matrice diagonale 4x4
y = torch.diagflat(x)
print(y)

Le résultat de sortie est:

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

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

Autres extensions