Pytorch常用函数用法归纳:创建tensor张量

1.torch.arange()

(1)函数原型

torch.arange(start,end,step,*,out,dtype,layout=,device,requires_grad)

(2)参数说明:

(3)函数功能:

返回大小为[(end-start)/step]的一维张量，其值为区间[start,end)中给定步长为step的等间隔取值

2.torch.range()

(1)函数原型:

torch.range(start,end,step,*,out,dtype,layout,device=None,requires_grad)

(2)参数说明:

(3)函数功能:

返回大小为[(end-start)/step]的一维张量，其值为区间[start,end]中给定步长为step的等间隔取值

3.生成随机数组成的Tensor张量:

(1)函数原型:

1.torch.rand(*size,generator,out,dtype,layout,device,requires_grad)
2.torch.randn(*size,generator,out,dtype,layout,device,requires_grad)
3.torch.randint(low,high,*size,generator,out,dtype,layout,device,requires_grad)
4.torch.randperm(n,generator,out,dtype,layout,device,requires_grad)
5.torch.normal(mean,std,size,generator,out)

(2)参数说明:

(3)函数功能:

1.生成由[0, 1)之间均匀分布的随机数组成的给定size大小的tensor张量;
2.生成由从标准正态分布(均值为0,标准差为1)中采样的随机数组成的指定size大小的tensor张量;
3.生成由指定范围[low,high)内的随机整数组成的指定size大小的tensor张量
4.生成一个从[0,n)的随机整数排列的tensor张量
5.生成由服从给定均值mean和标准差std的正态分布的随机数组成的指定size大小的tensor张量;
size参数不是必须的,当省略size参数时生成的tensor大小由mean和std决定

4.生成由固定值组成的Tensor张量 

(1)函数原型:

1.torch.ones(*size,out,dtype,layout,device,requires_grad)
2.torch.ones_like(input,dtype,layout,device,requires_grad)
3.torch.zeros(*size,out,dtype,layout,device,requires_grad)
4.torch.zeros_like(input,dtype,layout,device,requires_grad)
5.torch.fill(*size,fill_value,out,dtype,layout,device,requires_grad)
6.torch.eye(n,m,out,dtype,layout,device,requires_grad)
7.torch.empty(*size,out,dtype,layout,device,requires_grad)
8.torch.empty_like(input,out,dtype,layout,device,requires_grad)

(2)参数说明:

(3)函数功能:

1.生成给定size大小的一个全为1的Tensor张量
2.生成和输入张量input大小一致的一个全为1的Tensor张量
3.生成给定size大小的一个全为0的Tensor张量
4.生成和输入张量input大小一致的一个全为0的Tensor张量
5.生成给定size大小并且全为给定值fill_value的一个Tensor张量
6.生成大小为n*n的单位矩阵,是一个二维Tensor张量,默认情况下为方阵
7.生成给定size大小的一个填满未初始化数据的Tensor张量
8.生成和输入张量input大小一致的一个填满未初始化数据的Tensor张量

4.生成由现有数据值组成的Tensor张量 

(1)函数原型:

1.torch.tensor(data,dtype,device,requires_grad)
2.torch.from_numpy(ndarray)

(2)参数说明:

(3)函数功能:

1.根据给定的输入数据data创建一个指定大小的Tensor张量
2.根据给定的输入numpy数组创建一个指定大小的Tensor张量