Función torch.count_nonzero de PyTorch

Pytorch torch 参考手册Manual de referencia de Pytorch torch

torch.count_nonzeroEs una función en PyTorch que se utiliza para devolver el número de elementos distintos de cero en un tensor.

Definición de la función

torch.count_nonzero(input, dim)

Ejemplo de uso

Ejemplo

import torch

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

# Devuelve el número de elementos distintos de cero
print("Número de elementos distintos de cero:", torch.count_nonzero(x))

# Número de elementos distintos de cero a lo largo de dim=0
y = torch.tensor([[1, 0, 2], [0, 3, 0], [4, 0, 5]])
print("Número de elementos distintos de cero:", torch.count_nonzero(y))
print("dim=0 número de elementos distintos de cero:", torch.count_nonzero(y, dim=0))
print("dim=1 número de elementos distintos de cero:", torch.count_nonzero(y, dim=1))

El resultado es:

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

Pytorch torch 参考手册Manual de referencia de Pytorch torch

Otras extensiones