cr.sparse.lop.restriction

cr.sparse.lop.restriction(n, I, axis=0)[source]

An operator which computes y = x[I] over an index set I

Parameters
  • 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