PyTorch torch.nonzero-Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.nonzeroist eine Funktion in PyTorch, die verwendet wird, um die Indizes von Nicht-Null-Elementen zurückzugeben.

Funktionsdefinition

torch.nonzero(input)

Anwendungsbeispiele

Beispiel

import torch

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

# Die Indizes von Nicht-Null-Elementen zurückgeben
Indizes = torch.nonzero(x)

print("Indizes der Nicht-Null-Elemente:")
print(indices)

Das Ausgabeergebnis ist:

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

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen