test_convolutional.py 文件源码

def test_globalpooling_2d():
    layer_test(pooling.GlobalMaxPooling2D,
               kwargs={'dim_ordering': 'th'},
               input_shape=(3, 4, 5, 6))
    layer_test(pooling.GlobalMaxPooling2D,
               kwargs={'dim_ordering': 'tf'},
               input_shape=(3, 5, 6, 4))
    layer_test(pooling.GlobalAveragePooling2D,
               kwargs={'dim_ordering': 'th'},
               input_shape=(3, 4, 5, 6))
    layer_test(pooling.GlobalAveragePooling2D,
               kwargs={'dim_ordering': 'tf'},
               input_shape=(3, 5, 6, 4))