pdft.manifolds.batched_inv# pdft.manifolds.batched_inv(A)[source]# Batched matrix inverse via a transpose-and-invert trick. Reshape (d, d, n) → (n, d, d), call jnp.linalg.inv, reshape back. Mirror of upstream src/manifolds.jl:70-78 (but truly vectorized). Parameters: A (Array) Return type: Array