pytorch中常用函数总结

因为写代码的时候经常会忘记，网上搜比较麻烦，在这里自己记录一下。
（持续更新中）

#与numpy数组转换
a = torch.from_numpy(b)

#为数组某维度(0维度)添加一维
a.unsqueeze(0)

#交换维度,比如将[B, C, W, H]变换为[B, W, H, C]
a.permute(0, 2, 3, 1)

#连接向量,在第1个维度上
a = torch.cat([a, b], 1)

#加载网络
net = Net()
net.load_state_dict(torch.load(net_path))

#torchvision保存图片
torchvision.utils.save_image(img, save_path)