models.py 文件源码

python
阅读 22 收藏 0 点赞 0 评论 0

项目:sdp 作者: tansey 项目源码 文件源码
def get_density_probs(self):
        q = tf.matmul(self._input_layer, tf.transpose(self._W)) + self._b
        probs = tf.reciprocal(1 + tf.exp(-q)) # [batchsize, num_nodes]
        log_probs = tf.map_fn(lambda x: self._grid_log_probs(x), probs) # [batchsize, gridlen]
        return tf.exp(log_probs) / tf.reduce_sum(tf.exp(log_probs), axis=range(1,len(self._num_classes)+1), keep_dims=True)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号