1f9ad444b5dee8b2562b7bfa1cf9f576fa32a347,src/gluonnlp/op.py,,updated_vectors_by_position,#Any#Any#Any#Any#,99

Before Change

    // TODO(zheyuye), update when npx.index_update implemented
    updates = add_vectors_by_position(F, F.np.zeros_like(base), data, positions)
    updates_mask = add_vectors_by_position(F, F.np.zeros_like(base),
            F.np.ones_like(positions), positions)
    updates = (updates / F.np.maximum(1, updates_mask)).astype(np.int32)

    out = F.np.where(updates, updates, base)

After Change

    batch_idx = F.np.expand_dims(F.npx.arange_like(positions, axis=0),
                                 axis=1).astype(np.int32)
    batch_idx = batch_idx + F.np.zeros_like(positions)
    indices = F.np.stack([batch_idx.reshape(-1), positions.reshape(-1)])

    out = F.npx.index_update(base, indices, data.reshape(-1))
    return out

Instances