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)
    updates_mask = F.np.minimum(updates_mask, 1)

    return out, updates_mask


@use_np

After Change

        The updated result.
        Shape (batch_size, seq_length)
    
    positions = positions.astype(np.int32)
    // batch_idx.shape = (batch_size, 1) as [[0], [1], [2], ...]
    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