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

最新推荐文章于 2025-07-13 14:34:10 发布

小饼干超人

最新推荐文章于 2025-07-13 14:34:10 发布

阅读量6.6w

点赞数 53

CC 4.0 BY-SA版权

分类专栏： pytorch 文章标签： pytorch

本文链接：https://blog.youkuaiyun.com/m0_37586991/article/details/88830026

pytorch 专栏收录该内容

26 篇文章

订阅专栏

本文介绍了使用PyTorch创建两种不同类型的张量的方法：torch.range 和 torch.arange。前者生成包含指定结束值的浮点数张量，后者生成不包含指定结束值的整数张量。文中还对比了这两种函数生成的张量的数据类型差异。

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

>>> 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

总结：

torch.range(start=1, end=6) 的结果是会包含end的，
而torch.arange(start=1, end=6)的结果并不包含end。
两者创建的tensor的类型也不一样。

参考：
torch.range
torch.arange