PyTorch torch.nonzero Fonction

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

torch.nonzeroest une fonction de PyTorch utilisée pour renvoyer les indices des éléments non nuls.

Définition de fonction

torch.nonzero(input)

Exemples d'utilisation

Exemple

import torch

x = torch.tensor([[0, 1], [2, 0], [0, 3]])

# Renvoyer les indices des éléments non nuls
indices = torch.nonzero(x)

print("Indices des éléments non nuls :")
print(indices)

Le résultat de sortie est:

非零元素索引:
tensor([[0, 1],
        [1, 0],
        [2, 1]])

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

Autres extensions