def METHOD_NAME(self, input_shape):
    pooled_shape = 0
    for bin in self.bins:
        pooled_shape += tf.reduce_prod(bin)
    if self.data_format == "channels_last":
        return tf.TensorShape([input_shape[0], pooled_shape, input_shape[-1]])
    else:
        return tf.TensorShape([input_shape[0], input_shape[1], pooled_shape])