新人：请问为什么torch.arange(0,3)会输出整形 - pytorch吧 - RAT

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

level 2

gl19951106 楼主

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

2018年11月21日 13点11分 1

level 3

bf8268

路过
1.0以后的版本默认数据类型是float
之前的不确定，使用 torch.arange(0,3).float() 转换

2019年08月13日 03点08分 2

1