sparse.py 文件源码

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

项目:pytorch 作者: pytorch 项目源码 文件源码
def symbolic(g, indices, weight, padding_idx, max_norm, norm_type, scale_grad_by_freq,
                 sparse=False):
        if max_norm is not None:
            raise ValueError('Right now, re-norm is not supported.')

        return g.op("Gather", weight, indices)
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号