pytorc查看数据类型的方法是什么,哪些事项要注意
Admin 2022-06-29 群英技术资讯 481 次浏览
查看tensor数据大小时使用了data.shape(),报错:
TypeError: 'torch.Size' object is not callable 或 TypeError: 'tuple' object is not callable。
查看数据类型:data.dtype
查看数据大小:data.shape
补充:pytorch tensor比较大小 数据类型要注意
a = torch.tensor([[0, 0], [0, 0]]) print(a>=0.5)
输出
tensor([[1, 1], [1, 1]], dtype=torch.uint8)
结果明显不对, 分析原因是因为, a是long类型, 而0.5是float. 0.5会被转化为 long, 变为0. 因此结果会出错, 做出如下修改就可以得到正确答案
a = torch.tensor([[0, 0], [0, 0]]).float() print(a>=0.5)
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:mmqy2019@163.com进行举报,并提供相关证据,查实之后,将立刻删除涉嫌侵权内容。
猜你喜欢
这篇文章主要给大家分享pytorch函数的内容,本文给大家介绍两个函数,分别是squeeze函数、cat函数。那么这两个函数有什么用呢?用法是什么?下面我们一起来学习一下。
内容介绍函数的返回值函数的参数不可变参数和可变参数+=函数的参数缺省参数多值参数元组和字典的拆包总结函数的返回值一个函数执行后可以返回多个返回值defmeasure():print('测量开
大家好,本篇文章主要讲的是python实现简易猜数小游戏,感兴趣的同学赶快来看一看吧,对你有帮助的话记得收藏一下,方便下次浏览
想必大家都玩过贪吃蛇的游戏吧:通过操纵蛇的移动方向能够让蛇吃到随机出现的食物,吃到的食物越多,蛇就会变得越长。本文将使用手势识别来完成贪吃蛇这个简单的游戏,感兴趣的可以了解一下
这篇文章主要介绍了用Python的绘图库(matplotlib)绘制小波能量谱,代码简单详细,思路清晰,需要的朋友可以参考下
成为群英会员,开启智能安全云计算之旅
立即注册Copyright © QY Network Company Ltd. All Rights Reserved. 2003-2020 群英 版权所有
增值电信经营许可证 : B1.B2-20140078 粤ICP备09006778号 域名注册商资质 粤 D3.1-20240008