jax.numpy.diagflat ================== .. currentmodule:: jax.numpy .. autofunction:: diagflat