Contents:
An operator which computes y = x[I] over an index set I
n (int) – Dimension of model space
I (jax.numpy.ndarray) –
axis (int) – For multi-dimensional array input, the axis along which the linear operator will be applied