mm.math.gather

math.gather(indices, axis=None)

The values of the array at the given indices.

Parameters:

• array (ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]) – The array to gather values from.

• indices (Batch[int]) – The indices to gather values from.

• axis (Optional[int]) – The axis to gather values from.

Return type:

ndarray[Tuple[int, ...], Union[TypeVar(R, float16, float32, float64), TypeVar(C, complex64, complex128), TypeVar(Z, int16, int32, int64), TypeVar(N, uint16, uint32, uint64)]]

Returns:

The values of the array at the given indices.

code/api/mrmustard.math.gather

 

Download Python script

 

Download Notebook

 

View on GitHub