keras自带:
from keras.layers import Reshape
layer_1 = Reshape((height, width, chns))( layer1)
tensorflow中的reshape函数:
from keras import backend as K
K.reshape( layer1,(-1,2,4,8) )
keras自大的Reshape层不需要写batch的维度,但是tensorflow的reshape需要完整的维度。
keras自带:
from keras.layers import Reshape
layer_1 = Reshape((height, width, chns))( layer1)
tensorflow中的reshape函数:
from keras import backend as K
K.reshape( layer1,(-1,2,4,8) )
keras自大的Reshape层不需要写batch的维度,但是tensorflow的reshape需要完整的维度。