一、创建tensor张量
1.torch.arange()
(1)函数原型
torch.arange(start=0,end,step=1,*,out=None,dtype=None,layout=torch.strided,device=None,requires_grad=False)
(2)参数说明:
参数名称 | 参数类型 | 参数说明 |
start | Number | 起始值,默认值为0 |
end | Number | 结束值 |
step | Number | 步长值,默认为1 |
out | Tensor | 输出的张量 |
dtype | torch.dtype | 期望返回的张量的数据类型, |