pytorch中的resize_()和reshape()

a.resize_()会修改a的shape,a.reshape()返回修改后的tensor,但不会更改a的shape。

发布了51 篇原创文章 · 获赞 29 · 访问量 3万+

猜你喜欢

转载自blog.csdn.net/qwer7512090/article/details/104375141