test_keras_numeric.py 文件源码

def test_tiny_mcrnn_td(self):

        model = Sequential()
        model.add(Convolution2D(3,1,1, input_shape=(2,4,4), border_mode='same'))
        model.add(AveragePooling2D(pool_size=(2,2)))
        model.add(Reshape((2,3)))
        model.add(TimeDistributed(Dense(5)))

        self._test_keras_model(model)