def __init__(self, input, pool_shape=(2, 2), ignore_border=True,
activation_fn=None):
self.input = input
self.pool_shape = pool_shape
self.ignore_border = ignore_border
l_output = pool.pool_2d(input=input, ds=pool_shape,
ignore_border=ignore_border)
self.output = (l_output if activation_fn is None
else activation_fn(l_output))
评论列表
文章目录