test_keras2_numeric.py 文件源码

def test_tiny_spatial_bn(self):
        np.random.seed(1988)
        x_in = Input(shape=(7,7,2))
        x = ZeroPadding2D(padding=(1, 1))(x_in)
        x = BatchNormalization(axis=2)(x)
        model = Model(x_in, x)

        self._test_keras_model(model, input_blob = 'data', output_blob = 'output', delta=1e-2)