Pytorch torch.range()&torch.arange()

原创于 2020-10-23 09:10:42 发布 · 6.5k 阅读

17 ·

CC 4.0 BY-SA版权

Pytorch 专栏收录该内容

30 篇文章

订阅专栏

本文介绍了PyTorch中torch.range和torch.arange两个函数的使用区别。torch.range在创建一维张量时包含end值，其数据类型为float32，而torch.arange则不包含end值，返回的张量类型为int64。通过示例代码展示了两者不同的输出结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch.range(start=1, end=6)

结果是会包含end的，创建的tensor的类型为float32。

torch.range(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

torch.arange(start=1, end=6)

结果并不包含end，创建的tensor的类型为int64。

torch.arange(start=0, end, step=1, *, out=None, dtype=None, layout=torch.strided, device=None, requires_grad=False) → Tensor

代码

>>> y=torch.range(1,6)
>>> y
tensor([1., 2., 3., 4., 5., 6.])
>>> y.dtype
torch.float32

>>> z=torch.arange(1,6)
>>> z
tensor([1, 2, 3, 4, 5])
>>> z.dtype
torch.int64