search

facebookresearch / fmmax

Fourier modal method with Jax
MIT License
97 stars 10 forks source link

Eliminate batched 1D solve #119

Closed mfschubert closed 4 months ago

mfschubert commented 5 months ago

Eliminates the following warning:

FutureWarning: jnp.linalg.solve: batched 1D solves with b.ndim > 1 are deprecated, and in the future will be treated as a batched 2D solve. Use solve(a, b[..., None])[..., 0] to avoid this warning.