pytorc查看数据类型的方法是什么,哪些事项要注意
Admin 2022-06-29 群英技术资讯 346 次浏览
查看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一小时掌握之图像识别实战篇,本文给大家介绍的非常详细,对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下
目录题目描述示例 2:示例 3:单向构造(哈希表计数)双向构造(双指针)最后题目描述 这是 LeetCode 上的 1743. 从相邻元素对还原数组 ,难度为 中等。 Tag : 「哈希表」、「双指针」、「模拟」 存在一个由 n 个不同元素组成的整数数组 nums...
这篇文章主要介绍了Python中的简写操作(for、if简写、匿名函数),具有很好的参考价值,希望对大家有所帮助。如有错误或未考虑完全的地方,望不吝赐教
一些朋友应该有玩过吃豆人游戏,就是通过移动吃豆人吃豆豆来获得分数,同时要躲避怪物这样的玩法。小编认为是比较有趣的,因此给大家分享使用Python实现简单的吃豆人游戏代码,感兴趣的朋友可以参考学习。
这篇文章主要介绍了pygame实现井字棋之第二步逻辑实现,文中有非常详细的代码示例,对正在学习python的小伙伴们有非常好的帮助,需要的朋友可以参考下
成为群英会员,开启智能安全云计算之旅
立即注册Copyright © QY Network Company Ltd. All Rights Reserved. 2003-2020 群英 版权所有
增值电信经营许可证 : B1.B2-20140078 粤ICP备09006778号 域名注册商资质 粤 D3.1-20240008