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()

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

2.切分 -- torch.chunk() 和 torch.split()

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

张量索引

torch.index_select()  和 torch.masked_select()

Pytorch--张量操作

index必须是一个torch.long的数据类型

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

张量变换

torch.reshape() 和 torch.transpose()

重组形状,注意的一点就是重组后,数据量不能变

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

torch.squeeze() 和 torch.unsqueeze()

Pytorch--张量操作

如果本来就不为1的维度,就算你指定它,那也压缩不了。

Pytorch--张量操作

Pytorch--张量操作

张量运算

加减乘除

Pytorch--张量操作

Pytorch--张量操作

Pytorch--张量操作

================================资源推荐

本文课程截图主要来与深度之眼Pytorch训练营2期--张量操作部分,想了解更多更全更系统的课程请自行报名学习。

Pytorch--张量操作