Fonction torch.tril_indices de PyTorch

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

torch.tril_indicesC'est une fonction de PyTorch utilisée pour générer les indices d'une matrice triangulaire inférieure. Elle renvoie les indices de ligne et les indices de colonne de la partie triangulaire inférieure (y compris la diagonale).

Définition de fonction

torch.tril_indices(row, column, offset=0, dtype=torch.long, device='cpu')

Description des paramètres :

  • row: Nombre de lignes
  • column: Nombre de colonnes
  • offset: Décalage diagonal
  • dtype: Type de données de la valeur renvoyée
  • device: Périphérique

Exemple d'utilisation

Exemple

import torch

# Générer les indices triangulaires inférieurs pour une matrice 3x3
row, col = torch.tril_indices(3, 3)
print("row:", row)
print("col:", col)

Le résultat de sortie est:

row: tensor([0, 1, 1, 2, 2, 2])
col: tensor([0, 0, 1, 0, 1, 2])

Exemple

import torch

# Générer des indices et les utiliser pour les opérations d'indexation
row, col = torch.tril_indices(3, 3, offset=1)

# Créer une matrice 3x3
a = torch.ones(3, 3)

# Utiliser les indices pour définir la partie triangulaire inférieure
a[row, col] = 0
print(a)

Le résultat de sortie est:

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

Exemple

import torch

# Cas d'une matrice non carrée
row, col = torch.tril_indices(3, 4)
print("row:", row)
print("col:", col)

Le résultat de sortie est:

row: tensor([0, 1, 1, 2, 2, 2])
col: tensor([0, 0, 1, 0, 1, 2])

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

Autres extensions