np.newaxis

2019-06-25  本文已影响0人  你说你要一场

np.newaxis
可以为多维数组增加一个轴
比如:

使用形状60K[128*28]作为卷积网络的输入

X_train = X_train[:,np.newaxis,:,:]

将60K[2828] 转换为60K[128*28]

上一篇 下一篇

猜你喜欢

热点阅读