网络编程
位置:首页>> 网络编程>> Python编程>> pytorch 转换矩阵的维数位置方法

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

作者:zouxiaolv  发布时间:2021-05-21 03:48:15 

标签:pytorch,矩阵,维数

例如:


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

来源:https://blog.csdn.net/zouxiaolv/article/details/80936725

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com