PyTorch torch.count_nonzero 関数

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 リファレンスマニュアル

その他の拡張