e3x.ops.indexed.gather_src

e3x.ops.indexed.gather_src(inputs, *, adj_idx=None, src_idx=None, **_)

Gather from inputs (source) according to dense or sparse index list.

Parameters:

• inputs (<class 'Shaped[Array, '_*indexable_by_adj_idx_or_src_idx']'>) – Inputs to gather from.

• adj_idx (Optional[Integer[Array, '... N M']], default: None) – Adjacency indices (if dense index list is used), or None (if sparse index list is used).

• src_idx (Optional[Integer[Array, '... P']], default: None) – Source indices (if sparse index list is used), or None (if dense index list is used).

Return type:

<class 'Shaped[Array, '_*input_shape_indexed_by_adj_idx_or_src_idx']'>

Returns:

An array with the gathered values.

Raises:

RuntimeError – If neither dense nor sparse index lists are provided, or if both are provided.