Pytorch--张量操作
目录
1. torch.cat() 和 torch.stack()
2.切分 -- torch.chunk() 和 torch.split()
torch.index_select() 和 torch.masked_select()
torch.reshape() 和 torch.transpose()
torch.squeeze() 和 torch.unsqueeze()
张量操作
1. torch.cat() 和 torch.stack()
2.切分 -- torch.chunk() 和 torch.split()
张量索引
torch.index_select() 和 torch.masked_select()
index必须是一个torch.long的数据类型
张量变换
torch.reshape() 和 torch.transpose()
重组形状,注意的一点就是重组后,数据量不能变
torch.squeeze() 和 torch.unsqueeze()
如果本来就不为1的维度,就算你指定它,那也压缩不了。
张量运算
加减乘除
================================资源推荐
本文课程截图主要来与深度之眼Pytorch训练营2期--张量操作部分,想了解更多更全更系统的课程请自行报名学习。