如果y_hat是三维的,第一个维度是样本数,第二个维度是循环时间步,第三个维度是预测值,那么要怎么把第二个人维度循环时间步去掉,并且保持第一个维度和第二个维度不变,改过之后的y_hat是二维的

动态 未结 0 128
小小草
小小草 LV1 2023年9月23日 23:40 发表
可以使用numpy的reshape函数将三维数组转换为二维数组,其中第一个维度为原始数组的第一个维度,第二个维度为原始数组的第三个维度。代码如下: ```python import numpy as np # 假设y_hat是一个形状为(100, 10, 5)的三维数组 y_hat_reshaped = np.reshape(y_hat, (y_hat.shape[0], -1)) # y_hat_reshaped的形状为(100, 50),即去掉了第二个维度循环时间步 ``` 其中`-1`表示自动计算该维度的大小,以保证总元素数不变。因此,上述代码中`y_hat_reshaped`的第二个维度大小为原始数组的第二个维度大小乘以第三个维度大小。
收藏(0)  分享
相关标签: 智能问答
问题没解决?让AI助手帮你作答 AI助手
0个回复
  • 消灭零回复