最新下载
热门教程
- 1
- 2
- 3
- 4
- 5
- 6
- 7
- 8
- 9
- 10
pytorch tensor int型除法出现的问题
时间：2022-06-25 02:07:56 编辑：袖梨 来源：一聚教程网
本篇文章小编给大家分享一下pytorch tensor int型除法出现的问题，文章介绍的很详细，小编觉得挺不错的，现在分享给大家供大家参考，有需要的小伙伴们可以来看看。
accuracy对应的tensor出来是int型的，用到了一个除法取平均。而pytorch里无论用 / or // 结果都是取整，，accuracy一直就没有。。
所以转换成float就没问题了，，低级bug，下不为例
补充：pytorch tensor division/除法
除法的时候要注意数据类型的问题
a = torch.div(torch.tensor([160, 110]), 0.137)
jupyter notebook就会奔溃, 因为torch.tensor([160, 110])是torch.int64数据类型, 而0.137(或者说torch.tensor(0.137))是torch.float32类型.
如果改成下面这样, 就会顺利运行
a = torch.div(torch.tensor([160, 110]).float(), 0.137)
相关文章
- 崩坏星穹铁道娜塔莎如何培养 12-22
- 塞尔达无双封印战记最佳增强补给材料一览 增强补给材料图鉴 12-22
- 塞尔达无双封印战记最佳战斗补给材料一览 战斗补给材料图鉴 12-22
- 阴阳师天剑韧心鬼切有什么技能 12-22
- 塞尔达无双封印战记最佳支援补给材料一览 支援补给材料图鉴 12-22
- 崩坏星穹铁道米沙怎么养成 12-22