Fonction PyTorch torch.triu

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

torch.triuIl s'agit d'une fonction dans PyTorch utilisée pour extraire la partie triangulaire supérieure d'une matrice (y compris la diagonale principale). La partie triangulaire inférieure sera définie sur 0.

Définition de fonction

torch.triu(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

Exemples 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 supérieure
y = torch.triu(a)
print(y)

Le résultat de sortie est:

tensor([[1, 2, 3],
        [0, 5, 6],
        [0, 0, 9]])

Exemple

import torch

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

# Extraire la partie triangulaire supérieure située au-dessus de la première diagonale au-dessus de la diagonale principale
y = torch.triu(a, diagonal=1)
print(y)

Le résultat de sortie est:

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

Exemple

import torch

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

# Extraire la partie triangulaire supérieure
y = torch.triu(a)
print(y)

Le résultat de sortie est:

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

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

Autres extensions