• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    pytorch tensor int型除法出现的问题

    昨天晚上跑起来一个classification实验,今天发现训练loss在降,然而accuracy永远是0 。。。直觉告诉我evaluation有问题

    然后发现自己写了个很愚蠢的bug

    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)

    以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。如有错误或未考虑完全的地方,望不吝赐教。

    您可能感兴趣的文章:
    • Pytorch Tensor的索引与切片例子
    • Pytorch之扩充tensor的操作
    • pytorch 把图片数据转化成tensor的操作
    • Pytorch 扩展Tensor维度、压缩Tensor维度的方法
    • Pytorch生成随机数Tensor的方法汇总
    • PyTorch中Tensor的数据类型和运算的使用
    • Pytorch 使用tensor特定条件判断索引
    上一篇:pytorch单维筛选 相乘的案例
    下一篇:Python实现七大查找算法的示例代码
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    pytorch tensor int型除法出现的问题 pytorch,tensor,int,型,除法,