常见API
- view / reshape
- squeeze / unsqueeze
- transpose / t / permute
- expand / repeat
View / Reshape



- View 与 Reshape的区别
1、reshape()
可以torch.reshape()
, 还可以torch.Tensor.reshape
;view()
只能torch.Tensor.view()
2、对于一个将要被view的Tensor,新的size必须与原来的size和stride兼容。否则,在view之前必须调用contiguous()方法。对于这一点,我还不能解释完全清楚。
会发现a.stride() != b.stride()
,总之在使用view()
方法时候报错上图所示,就先使用一下contiguous()
方法。
网友评论