range arange

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/sinat_26114733/article/details/88069318

这是一个小坑,
torch.range 已经被弃用了/

torch.range(0,5)
__main__:1: UserWarning: torch.range is deprecated in favor of torch.arange and will be removed in 0.5. Note that arange generates values in [start; end), not [start; end].
tensor([0., 1., 2., 3., 4., 5.])

因为不符合直接,真是左闭右闭的区间。而且是float。

torch.arange(0,5)
tensor([0, 1, 2, 3, 4])

猜你喜欢

转载自blog.csdn.net/sinat_26114733/article/details/88069318