relife.utils.args_take
relife.utils.args_take¶
- relife.utils.args_take(indices: numpy.ndarray, *args) Tuple[numpy.ndarray, ...][source]¶
Take elements in each array of args on axis=-2.
- Parameters
indices (ndarray) – The indices of the values to extract for each array.
*args (float or 2D array) – Sequence of arrays.
- Returns
The tuple of arrays where values at indices are extracted.
- Return type
Tuple[ndarray, …]