torch.linspace，unsqueeze以及squeeze函数
1.torch.linspace(start,end,steps=100，dtype)
作用是返回一个一维的tensor(张量）,其中dtype是返回的数据类型。
import torch print(torch.linspace(-1,1,5))
输出结果为：tensor([-1.0000, -0.5000, 0.0000, 0.5000, 1.0000])
2.unsqueeze()函数
在指定位置增加维度。
import torch a=torch.arange(0,6) #a是一维向量 b=a.reshape(2,3) #b是二维向量 c=b.unsqueeze(1) #c是三维向量，在b的第二维上增加一个维度 print(a) print(b) print(c) print(c.size())
a的维度为1x6
b的维度为2x3
b的维度为2x1x3
若想在倒数第二个维度增加一个维度，则c=b.unsqueeze(-1)
3.squeeze()函数
可去掉维度为1的维度。
import torch a=torch.arange(0,6) #a是一维向量 b=a.reshape(2,3) c=b.unsqueeze(1) print(c) print(c.size()) d=c.squeeze(1) print(d) print(d.size())
输出结果为：
优质内容筛选与推荐>>1、【bzoj3343】教主的魔法 (分块 + 二分)
2、接口 interface 以及适配器设计模式原理
3、懂，你的App生，不懂，死！
4、安装配置Apache2.4和php7.0
5、关于oracle的相关基础语句