pyTorch中torch.range()和torch.arange()的区别

最新推荐文章于 2022-06-06 21:16:48 发布

一颗苹果OAOA

最新推荐文章于 2022-06-06 21:16:48 发布

阅读量2.5k

点赞数 2

分类专栏： pyTorch深度学习框架文章标签： python 深度学习

本文链接：https://blog.csdn.net/qq_43360533/article/details/107335389

版权

pyTorch深度学习框架专栏收录该内容

30 篇文章 10 订阅

订阅专栏

torch.range()和torch.arange()的区别

x = torch.range(-8, 8)
y = torch.arange(-8, 8)
print(x, x.dtype)
print(y, y.dtype)

output：
tensor([-8., -7., -6., -5., -4., -3., -2., -1., 0., 1., 2., 3., 4., 5.,6., 7., 8.]) torch.float32
tensor([-8, -7, -6, -5, -4, -3, -2, -1, 0, 1, 2, 3, 4, 5, 6, 7]) torch.int64
可以看到，torch.range()的范围是[-8, 8]，类型为torch.float32
torch.arange()的范围是[-8, 8)，类型为torch.int64

在梯度设置时会出现错误：

x = torch.range(-8, 8, 1, requires_grad=True)
y = torch.arange(-8, 8, 1, requires_grad=True)
print(x, x.dtype)
print(y, y.dtype)

在这里插入图片描述
即只有当类型为float时才可设置requires_grad=True，故可将

y = torch.arange(-8, 8, 1, requires_grad=True)

改为以下，即手动改变数据类型即可。

y = torch.arange(-8.0, 8.0, 1.0, requires_grad=True)

output：
tensor([-8., -7., -6., -5., -4., -3., -2., -1., 0., 1., 2., 3., 4., 5.,6., 7., 8.], requires_grad=True)
torch.float32
tensor([-8., -7., -6., -5., -4., -3., -2., -1., 0., 1., 2., 3., 4., 5.,6., 7.], requires_grad=True)
torch.float32

欢迎关注【OAOA】