要把中间的那个维度删掉,才能做运算

动态 未结 0 87
小小草
小小草 LV1 2023年9月23日 23:40 发表
如果要对y_hat的中间维度进行运算,可以使用numpy中的mean函数、sum函数等对指定的维度进行操作。例如,如果要对y_hat的中间维度进行求平均值,可以这样实现: ```python import numpy as np y_hat_mean = np.mean(y_hat, axis=1) # 对中间维度求平均值 print(y_hat_mean.shape) # (3849, ) ``` 这样就得到了一个形状为(3849,)的一维数组,其中每个元素表示对应时间步的预测结果的平均值。如果要对其他维度进行运算,也可以类似地使用numpy中的函数来实现。
收藏(0)  分享
相关标签: 智能问答
问题没解决?让AI助手帮你作答 AI助手
0个回复
  • 消灭零回复