crbm.py 文件源码

def __pool(self):
        # Pooling
        with tf.name_scope('pooling') as _:
            self.pooled_prob = tf.sub(1., tf.div(1., 1. + tf.mul(self.__avg_pool(tf.exp(self.signal)), self.pool_size * self.pool_size)), name='pooled_prob')
            self.pooled = self.__sample(self.pooled_prob, 'pooled')