代码驿站移动版

频道导航

pytorch 转换矩阵的维数位置方法

时间：2022-03-05 09:51:08 | 栏目：Python代码 | 点击：次

例如：

preds = to_numpy(preds)#preds是[2985x16x2]
preds = preds.transpose(2, 1, 0)#preds[2x16x2985]

您可能感兴趣的文章:

相关文章