View在pytorch中view函数的作用为重构张量的维度,相当于numpy中resize()的功能,但是用法可能不太一样 ... Tensor(128,1,512) B = a.repeat(1,5,1) print(B.shape).
確定! 回上一頁