[IT]/python.numpy

np.moveaxis

givemebro 2020. 5. 13. 16:10
반응형
# 3 X 1080 X 1920 -> 1080 X 1920 X 3(moveaxis)

X_train=np.moveaxis(X_train.reshape(-1,3,32,32),1,-1)
반응형