PyTorch torch.is_nonzero 関数

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

torch.is_nonzeroPyTorch において、テンソルが非ゼロの単一要素であるかどうかを確認するための関数です。

関数定義

torch.is_nonzero(input)

使用例

実例

import torch

print(torch.is_nonzero(torch.tensor([1.0])))
print(torch.is_nonzero(torch.tensor([0.0])))
print(torch.is_nonzero(torch.tensor([1, 2])))

Pytorch torch 参考手册Pytorch torch リファレンスマニュアル

その他の拡張