torch.is_nonzero(input)
torch.is_nonzero(input)
- input: 输入张量
若输入是 不等于零的单元素张量 则返回True，否则返回False
- 不等于零的单元素张量：
torch.tensor([0.])或torch.tensor([0])或torch.tensor([False]) - 单元素张量: 只有一个数 的张量
import torch
print(torch.is_nonzero(torch.tensor([0.]))) # False
print(torch.is_nonzero(torch.tensor([1.5])))# True
print(torch.is_nonzero(torch.tensor([False])))# False
print(torch.is_nonzero(torch.tensor([3])))# True
# torch.is_nonzero(torch.tensor([1, 3, 5])) # 有三个元素，不是单元素张量
# 报错：RuntimeError: Boolean value of Tensor with more than one value is ambiguous
# torch.is_nonzero(torch.tensor([])) #没有元素，不是单元素张量
# 报错：RuntimeError: Boolean value of Tensor with no values is ambiguous
原文地址：https://blog.csdn.net/m0_63070489/article/details/144041873
免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！