pytorch.range()和pytorch.arange()的区别及说明

更新时间：2023年08月03日 14:28:37 作者：小饼干超人

这篇文章主要介绍了pytorch.range()和pytorch.arange()的区别及说明,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

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

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

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

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

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

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

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

代码

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

pytorch torch.arange

应用

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

API

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

参数	描述
start (Number)
end (Number)
step (Number)
out (Tensor, optional)
dtype (torch.dtype, optional)
layout (torch.layout, optional)
device (torch.device, optional)
requires_grad (bool, optional)

总结

以上为个人经验，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

tensorflow基于CNN实战mnist手写识别(小白必看)
这篇文章主要介绍了tensorflow基于CNN实战mnist手写识别(小白必看),文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
2020-07-07
Python使用chardet判断字符编码
这篇文章主要介绍了Python使用chardet判断字符编码的方法,较为详细的分析了Python中chardet的功能、安装及使用技巧,需要的朋友可以参考下
2015-05-05
Python远程开发环境部署与调试过程图解
这篇文章主要介绍了Python远程开发环境部署与调试过程图解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
2019-12-12
Python设计模式之MVC模式简单示例
这篇文章主要介绍了Python设计模式之MVC模式,简单介绍了MVC模式的概念、原理,并结合实例形式给出了Python实现与使用MVC模式的相关操作技巧,需要的朋友可以参考下
2018-01-01
Python之串口收发的异步程序
这篇文章主要介绍了Python之串口收发的异步程序,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教
2023-09-09
python变量命名的7条建议
这篇文章主要介绍了python变量命名的7条建议，如何给变量命名，如何让它变得有意义成了程序员不可逾越的难题,需要的朋友可以参考下
2019-07-07
pygame实现俄罗斯方块游戏（对战篇1）
这篇文章主要为大家详细介绍了pygame实现俄罗斯方块游戏的对战篇，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2019-10-10
pymssql数据库操作MSSQL2005实例分析
这篇文章主要介绍了pymssql数据库操作MSSQL2005的方法,可实现基本的连接、查询、插入、更新及调用存储过程等功能,非常具有实用价值,需要的朋友可以参考下
2015-05-05
python selenium禁止加载某些请求的实现
本文主要介绍了python selenium禁止加载某些请求的实现，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2022-01-01
python+html文字点选验证码加固安全防线
这篇文章主要为大家介绍了python文字点选验证码加固安全防线实现示例详解,有需要的朋友可以借鉴参考下,希望能够有所帮助,祝大家多多进步,早日升职加薪
2023-09-09