torch.is_nonzero(input)

torch.is_nonzero(input)

  • input: 输入张量

若输入是 不等于零的单元素张量 则返回True,否则返回False

  • 不等于零的单元素张量:torch.tensor([0.])torch.tensor([0])torch.tensor([False])
  • 单元素张量: 只有一个数 的张量
python 复制代码
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
相关推荐
白熊1882 小时前
【大模型LLM】梯度累积(Gradient Accumulation)原理详解
人工智能·大模型·llm
愚戏师2 小时前
机器学习(重学版)基础篇(算法与模型一)
人工智能·算法·机器学习
仰望星空的凡人3 小时前
【JS逆向基础】数据库之MongoDB
javascript·数据库·python·mongodb
F_D_Z3 小时前
【PyTorch】图像多分类项目部署
人工智能·pytorch·python·深度学习·分类
pingzhuyan4 小时前
python入门篇12-虚拟环境conda的安装与使用
python·ai·llm·ocr·conda
香蕉可乐荷包蛋4 小时前
排序算法 (Sorting Algorithms)-Python示例
python·算法·排序算法
音视频牛哥5 小时前
打通视频到AI的第一公里:轻量RTSP服务如何重塑边缘感知入口?
人工智能·计算机视觉·音视频·大牛直播sdk·机器视觉·轻量级rtsp服务·ai人工智能
Wendy14416 小时前
【灰度实验】——图像预处理(OpenCV)
人工智能·opencv·计算机视觉
中杯可乐多加冰6 小时前
五大低代码平台横向深度测评:smardaten 2.0领衔AI原型设计
人工智能
无线图像传输研究探索6 小时前
单兵图传终端:移动场景中的 “实时感知神经”
网络·人工智能·5g·无线图传·5g单兵图传