keras中的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需要完整的维度。

猜你喜欢

转载自blog.csdn.net/moshiyaofei/article/details/87888451