解决方案:
import numpy as np
np.squeeze(Y)
np.squeeze(Y.train)
np.squeeze这个函数的作用是去掉矩阵里维度为1的维度。...例:(1, 300)的矩阵经由np.squeeze处理后变成300; (200, 1, 300)的矩阵经由np.squeeze处理后变成(200, 300)。...Examples
>>>
>>> x = np.array([[[0], [1], [2]]])
>>> x.shape
(1, 3, 1)
>>> np.squeeze(x).shape
(3,)
>...>> np.squeeze(x, axis=(2,)).shape
(1, 3)