pdft.manifolds.batched_inv

Contents

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