Pytorch常用函数用法归纳:创建tensor张量
1.torch.arange()
（图片来源网络，侵删）
(1)函数原型
torch.arange(start,end,step,*,out,dtype,layout=,device,requires_grad)
(2)参数说明:
| 参数名称 | 参数类型 | 参数说明 |
| start | Number | 起始值，默认值为0 |
| end | Number | 结束值，取不到，为开区间 |
| step | Number | 步长值，默认为1 |
| out | Tensor | 输出的张量，，表明创建后tensor赋予哪个变量，通常情况下不会设置该参数 |
| dtype | torch.dtype | 期望返回的张量的数据类型，如果是None,则使用全局默认值，如果未给出dtype,则从其他输入参数推断数据类型，如果start、end或stop中的任何一个是浮点数，则dtype被推断为默认值，否则被推断为torch.int64 |
| layout | torch.layout | 返回张量的期望 layout,默认值为torch.strided |
| device | torch.device | 返回张量的期望设备。如果是默认值None，则使用当前设备作为默认张量类型，对于CPU类型的张量，则device是CPU;若是CUDA 类型的张量，则device是当前的CUDA 设备 |
| requires_grad | bool | autograd是否记录返回张量上的梯度。默认值为False，表示不记录 |
(3)函数功能:
返回大小为[(end-start)/step]的一维张量，其值为区间[start,end)中给定步长为step的等间隔取值
2.torch.range()
(1)函数原型:
torch.range(start,end,step,*,out,dtype,layout,device=None,requires_grad)
(2)参数说明:
| 参数名称 | 参数类型 | 参数说明 |
| start | Number | 起始值，默认值为0 |
| end | Number | 结束值，可以取到，为闭区间 |
| step | Number | 步长值，默认为1 |
| out | Tensor | 输出的张量，表明创建后tensor赋予哪个变量，通常情况下不会设置该参数 |
| dtype | torch.dtype | 期望返回的张量的数据类型，如果是None,则使用全局默认值;如果未给出dtype,则从其他输入参数推断数据类型;如果start、end或stop中的任何一个是浮点数，则dtype被推断为默认值;否则被推断为torch.int64 |
| layout | torch.layout | 返回张量的期望layout,默认值为torch.strided,对性能影响不大 |
| device | torch.device | 返回张量的设备。默认值None表示使用当前设备作为默认张量类型;对于CPU类型的张量则device是CPU;若是CUDA 类型的张量则device是当前的CUDA 设备 |
| requires_grad | bool | autograd是否记录返回张量上的梯度。默认值为False，表示不记录 |
(3)函数功能:
返回大小为[(end-start)/step]的一维张量，其值为区间[start,end]中给定步长为step的等间隔取值
免责声明：我们致力于保护作者版权，注重分享，被刊用文章因无法核实真实出处，未能及时与作者取得联系，或有版权异议的，请联系管理员，我们会立即处理! 部分文章是来自自研大数据AI进行生成,内容摘自(百度百科,百度知道,头条百科,中国民法典,刑法,牛津词典,新华词典,汉语词典,国家院校,科普平台)等数据,内容仅供学习参考,不准确地方联系删除处理! 图片声明：本站部分配图来自人工智能系统AI生成,觅知网授权图片,PxHere摄影无版权图库和百度，360，搜狗等多加搜索引擎自动关键词搜索配图，如有侵权的图片，请第一时间联系我们，邮箱：ciyunidc@ciyunshuju.com。本站只作为美观性配图使用,无任何非法侵犯第三方意图,一切解释权归图片著作权方,本站不承担任何责任。如有恶意碰瓷者,必当奉陪到底严惩不贷!
