pytorch基本语法总结【持续更新】

1.np.arange()
只需记住第一个是起始点，第二个终点，第三个是步长，终点必须有，起始点默认为0，步长默认为1

#一个参数 默认起点0，步长为1
a = np.arange(3)

#两个参数 默认步长为1
a = np.arange(5,8)

#三个参数 起点为0，终点为5，步长为0.1
a = np.arange(0, 5, 0.1)

2.unsqueeze()函数起升维的作用,参数表示在哪个地方加一个维度。对于tensor而言

a.unsqueeze(dim=0)

在dim=0维度加1维
3，相应的，squeeze()函数起降维的作用