出现这样的情况你因为你矩阵的维度出现了冗余情况,比如你把一组数放入矩阵,矩阵默认的维度是2,但是你其实只有一列数,
或者你实际是2维的数据,你将其转为3维数据形式,也会报错。
因此可以先用np.shape函数查看你的矩阵维度,是否出现了(n,)这样的情况。
然后对矩阵进行reshape重构,或者np.squeeze去除冗余自由度就可以避免这样的问题。
data = np.reshape(data,[-1,20,28,28])
or
Y_prediction_test = np.squeeze(d['Y_prediction_test'])