pytorch每日一学5(torch.is_nonzero(input))

最新推荐文章于 2024-01-02 11:44:36 发布

Fluid_ray

最新推荐文章于 2024-01-02 11:44:36 发布

阅读量855

点赞数 2

分类专栏： pytorch每日一学文章标签：深度学习 pytorch 数据挖掘机器学习神经网络

本文链接：https://blog.csdn.net/Fluid_ray/article/details/109477263

版权

49 篇文章 258 订阅

订阅专栏

5.第五个方法

torch.is_nonzero(input)

这个方法虽然看起来好像很难直观，但是其实这个方法的用途可能和大家想的不太一样。

此方法的用途是测试一个单元素的tensor在进行类型转化后是不是为0。就是测试input是不是不为torch.tensor([0.])或者torch.tensor([0])或者torch.tensor([False])三个其中的一个。
如图所示，测试结果如上所示。
如果输入tensor长度不为1则会返回一个RuntimeError，如下所示：
即使tensor长度为空也是一样的。