e3x.ops.indexed.sparse_to_dense_indices

e3x.ops.indexed.sparse_to_dense_indices(dst_idx, src_idx, pad_idx=None)

Converts a sparse index list to a dense index list.

Note: Not compatible with jax.jit or jax.vmap because boolean arrays are not concrete.

Parameters:

• dst_idx (<class 'Integer[Array, '... P']'>) – Destination indices.

• src_idx (<class 'Integer[Array, '... P']'>) – Source indices.

• pad_idx (Optional[int], default: None) – All indices >= pad_idx are assumed to be padding. If this is None, all indices are assumed to be non-padding entries.

Return type:

<class 'Integer[Array, '... N M']'>

Returns:

The corresponding dense index list.