networks.py 文件源码

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

项目:comprehend 作者: Fenugreek 项目源码 文件源码
def init_train_args(self, mode='recode'):
        # To be used for training by tf.Optimizer objects.
        self.train_args = [tf.placeholder(self.dtype,
                                          shape=[None, self.n_visible])]
        if mode == 'target':
            self.train_args.append(tf.placeholder(self.dtype,
                                                  shape=[None, self.n_hidden]))
        elif mode == 'label':
            self.train_args.append(tf.placeholder(tf.int32, shape=[None]))

        return self.train_args
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号