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)
评论列表
文章目录