pytorch中的reshape和resize
时间: 2023-04-06 10:01:24 浏览: 300
pytorch之Resize()函数具体使用详解
函数有什么区别?
reshape和resize函数都可以用来改变张量的形状,但是它们的实现方式不同。reshape函数会返回一个新的张量,而resize函数会直接修改原始张量的形状。此外,reshape函数可以自动推断某个维度的大小,而resize函数需要手动指定每个维度的大小。
阅读全文
pytorch之Resize()函数具体使用详解