pymc.math.batched\_diag ======================= .. currentmodule:: pymc.math .. autofunction:: batched_diag