ee4ce8e9d9a439b9a87fcfa6abdcb0f7f3f62b11,torch/onnx/symbolic_opset9.py,,embedding,#Any#Any#Any#Any#Any#Any#,474

Before Change

def embedding(g, weight, indices, padding_idx, scale_grad_by_freq, sparse):
    return g.op("Gather", weight, indices)


@parse_args("v", "v", "v", "i", "i", "i", "v", "i")
def embedding_bag(g,

After Change

    // auto out = weight.index_select(0, indices.reshape(-1));
    // zerofill_padding(out);
    // return out.view(size);
    weight = g.op("Gather", weight, indices)
    if (padding_idx >= 0):
        mask = eq(g, indices, g.op("Constant", value_t=torch.tensor(padding_idx)))
        if sym_help._export_onnx_opset_version < 11:
            mask = unsqueeze(g, mask, -1)

Instances