numpy 使用 transpose
pytorch使用permute, 虽然拥有transpose,但是只可以交换两维数据
numpy交换维度以及pytorch交换维度
猜你喜欢
转载自blog.csdn.net/weixin_36149892/article/details/85220032
今日推荐
周排行