pytorch

pytorch学习第一天

pytorch是一款深度学习框架，对模型进行加速训练。

pytorch的tensor

pytorch声明tensor有几种方式

torch.tensor() #里面可以是list，也可以是一个单个的数字。
torch.Tensor() #括号里面声明的是tensor的shape，也可以是list。如果是list的话tensor声明的是list里面的内容。
torch.from_numpy() #将numpy里面的内容转换为tensor
torch.rand(2,3)# 声明一个2行3列符合均匀分布的tensor
torch.randn(2,3)# 声明一个2行3列符合正态分布的tensor

pytorch查看维度

a = torch.tensor([1.1,2.2])
a.dim()=len(a.shape)
a.shape
a.size # 返回类型为torch.size的类型可以强转为list类型

tensor增加维度

a.unsqueeze(0)# 在0维之前新增一个维度
a.unsqueeze(-1)# 在-1维之后新增一个维度
a.squeeze(1)# 删除1维。

tensor扩张维度

a = torch.randn(32)
# 生成dimision为1，长度为32的tensor
a.unsqueeze(0).unsqueeze(1).unsqueeze(3).shape
a.expand(4，3，32，28) #对维度为1的维度进行扩张,如果生成的是4维他的范围就是[-3,4]
a,expand(-1,-1,-1,-1)# -1表示这个维度不变

tensor对维度改变

a.view()
a.transpose(1,3)
a.permute(4,3,2,1)