Функция torch.count_nonzero в PyTorch

Pytorch torch 参考手册Справочное руководство PyTorch torch

torch.count_nonzeroявляется функцией в PyTorch, используемой для возврата количества ненулевых элементов в тензоре.

Определение функции

torch.count_nonzero(input, dim)

Пример использования

Пример

import torch

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

# Возвращает количество ненулевых элементов
print("Количество ненулевых элементов:", torch.count_nonzero(x))

# Количество ненулевых элементов вдоль dim=0
y = torch.tensor([[1, 0, 2], [0, 3, 0], [4, 0, 5]])
print("Количество ненулевых элементов:", torch.count_nonzero(y))
print("Количество ненулевых элементов вдоль dim=0:", torch.count_nonzero(y, dim=0))
print("Количество ненулевых элементов вдоль dim=1:", torch.count_nonzero(y, dim=1))

Результат вывода:

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

Pytorch torch 参考手册Справочное руководство PyTorch torch

Другие расширения