Contents:
Returns a linear operator which mimics multiplication by a diagonal matrix
d (jax.numpy.ndarray) – A vector (1D array) of diagonal entries
axis (int) – For multi-dimensional array input, the axis along which the linear operator will be applied
A linear operator wrapping the diagonal matrix multiplication
Operator