Fonction torch.count_nonzero de PyTorch

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

torch.count_nonzeroest une fonction dans PyTorch utilisée pour renvoyer le nombre d'éléments non nuls dans un tenseur.

Définition de fonction

torch.count_nonzero(input, dim)

Exemple d'utilisation

Exemple

import torch

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

# Renvoyer le nombre d'éléments non nuls
print("Nombre d'éléments non nuls :", torch.count_nonzero(x))

# Nombre d'éléments non nuls selon dim=0
y = torch.tensor([[1, 0, 2], [0, 3, 0], [4, 0, 5]])
print("Nombre d'éléments non nuls :", torch.count_nonzero(y))
print("Nombre d'éléments non nuls selon dim=0 :", torch.count_nonzero(y, dim=0))
print("Nombre d'éléments non nuls selon dim=1 :", torch.count_nonzero(y, dim=1))

Le résultat de sortie est:

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

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

Autres extensions