pytorch转换矩阵的维数位置

  • Post author:
  • Post category:其他


例如:

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




版权声明:本文为zouxiaolv原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。