PyTorch torch.is_nonzero 函数

Pytorch torch 参考手册 Pytorch torch 参考手册

torch.is_nonzero 是 PyTorch 中用于检查张量是否为非零单一元素的函数。

函数定义

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 参考手册