Python-view修改矩阵形状

目录

 

1、简介

2、使用

3、示例

1、简介

最近看到代码中出现了很多次的view操作，经过学习发现view()的作用相当于numpy中的reshape，重新定义矩阵的形状。

参考

• PyTorch中view的用法
• PyTorch中的view的用法

2、使用

假如对X(m,n)指定两个参数的话：torch.view(参数a,参数b)

• 当a和b都指定的话，就是转换的形状【a,b】，需要注意的是转换前后两个维度相乘大小不变,也就是m*n=a*b