Just generalising the above solution for any upsampling factor 'r' like in pixel shuffle. B = A.reshape(-1,r,3,s,s).permute(2,3,0,4,1).reshape(1,3,rs,rs) ...
確定! 回上一頁