pytorch 转换矩阵的维数位置方法
时间:2022-03-05 09:51:08|栏目:Python代码|点击: 次
例如:
preds = to_numpy(preds)#preds是[2985x16x2] preds = preds.transpose(2, 1, 0)#preds[2x16x2985]
上一篇:Python面向对象程序设计之类的定义与继承简单示例
栏 目:Python代码
本文标题:pytorch 转换矩阵的维数位置方法
本文地址:http://www.codeinn.net/misctech/195306.html