model.py 文件源码

def down_block(block_fn, filters):
    def f(inputs):
        x = block_fn(filters)(inputs)
        x = block_fn(filters)(x)
        down = tcl.max_pool2d(x,
                              kernel_size = (3, 3),
                              stride = (2, 2),
                              padding = 'SAME')
        return x, down # x:same size of inputs, down: downscaled
    return f