PyTorch torch.count_nonzero-Funktion

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

torch.count_nonzeroist eine Funktion in PyTorch, die verwendet wird, um die Anzahl der Nicht-Null-Elemente in einem Tensor zurückzugeben.

Funktionsdefinition

torch.count_nonzero(input, dim)

Anwendungsbeispiel

Beispiel

import torch

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

# Gibt die Anzahl der Nicht-Null-Elemente zurück
print("Anzahl der Nicht-Null-Elemente:", torch.count_nonzero(x))

# Anzahl der Nicht-Null-Elemente entlang dim=0
y = torch.tensor([[1, 0, 2], [0, 3, 0], [4, 0, 5]])
print("Anzahl der Nicht-Null-Elemente:", torch.count_nonzero(y))
print("Anzahl der Nicht-Null-Elemente entlang dim=0:", torch.count_nonzero(y, dim=0))
print("Anzahl der Nicht-Null-Elemente entlang dim=1:", torch.count_nonzero(y, dim=1))

Das Ausgabeergebnis ist:

非零元素数量: tensor(4)
非零元素数量: tensor(5)
dim=0 非零元素数量: tensor([2, 1, 2])
dim=1 非零元素数量: tensor([2, 1, 2])

Pytorch torch 参考手册PyTorch torch-Referenzhandbuch

Andere Erweiterungen