Fonction torch.tril de PyTorch

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

torch.trilest une fonction dans PyTorch utilisée pour extraire la partie triangulaire inférieure d'une matrice (y compris la diagonale principale). La partie triangulaire supérieure sera mise à 0.

Définition de fonction

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

Description des paramètres :

  • input: tenseur d'entrée
  • diagonal: indice de diagonale, 0 indique la diagonale principale
  • out: tenseur de sortie

Exemple d'utilisation

Exemple

import torch

# Créer une matrice
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Extraire la partie triangulaire inférieure
y = torch.tril(a)
print(y)

La sortie est :

tensor([[1, 0, 0],
        [4, 5, 0],
        [7, 8, 9]])

Exemple

import torch

# Créer une matrice
a = torch.tensor([[1, 2, 3], [4, 5, 6], [7, 8, 9]])

# Extraire la partie triangulaire inférieure à partir de la première diagonale sous la diagonale principale
y = torch.tril(a, diagonal=1)
print(y)

La sortie est :

tensor([[1, 2, 0],
        [4, 5, 6],
        [7, 8, 9]])

Exemple

import torch

# Créer une matrice non carrée
a = torch.tensor([[1, 2, 3], [4, 5, 6]])

# Extraire la partie triangulaire inférieure
y = torch.tril(a)
print(y)

La sortie est :

tensor([[1, 0, 0],
        [4, 5, 0]])

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

Autres extensions