View在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能,但是用法可能不 ... 1, 28, 28)print(a.shape)print(a.view(4, 28 * 28).shape)print(a.
確定! 回上一頁