例如:
preds = to_numpy(preds)#preds是[2985x16x2]
preds = preds.transpose(2, 1, 0)#preds[2x16x2985]
版权声明:本文为zouxiaolv原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。