【水】pytorch:torch.reshape和torch.Tensor.view的区别

【水】pytorch:torch.reshape和torch.Tensor.view的区别

注:本篇仅为学习笔记,请谨慎参考,如有错误请评论指出。
参考:Pytorch: view()和reshape()的区别?他们与continues()的关系是什么?

两者区别

torch.reshape可能会返回一个从原张量复制而来的新张量,torch.Tensor.view会直接操作内存中的张量,只有torch.reshape满足torch.Tensor.view的条件时才会跟view一样直接操作内存中的张量。

1、使用torch.Tensor.view得到的张量与原张量共享内存,如果原张量的值变换那么新张量也会变化。
2、torch.Tensor.view只有在满足连续性的条件下才能使用。

你可能感兴趣的:(pytorch,人工智能,python)