UpdateFunction.py 文件源码

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

项目:nmp_qc 作者: priba 项目源码 文件源码
def u_intnet(self, h_v, m_v, opt):
        if opt['x_v'].ndimension():
            input_tensor = torch.cat([h_v, opt['x_v'], torch.squeeze(m_v)], 1)
        else:
            input_tensor = torch.cat([h_v, torch.squeeze(m_v)], 1)

        return self.learn_modules[0](input_tensor)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号