pytorch吧 关注:1,457贴子:2,583
  • 1回复贴,共1

新人:请问为什么torch.arange(0,3)会输出整形

只看楼主收藏回复

新人:请问为什么torch.arange(0,3)会输出整形Tensor


来自iPhone客户端1楼2018-11-21 21:59回复
    路过
    1.0以后的版本默认数据类型是float
    之前的不确定,使用 torch.arange(0,3).float() 转换


    IP属地:河南2楼2019-08-13 11:53
    回复