diff --git a/exponax/nonlin_fun/_convection.py b/exponax/nonlin_fun/_convection.py index 39f4aca..732b173 100644 --- a/exponax/nonlin_fun/_convection.py +++ b/exponax/nonlin_fun/_convection.py @@ -131,7 +131,7 @@ def _multi_channel_conservative_eval( u = self.ifft(u_hat_dealiased) u_outer_product = u[None, :] * u[:, None] u_outer_product_hat = self.fft(u_outer_product) - convection = jnp.sum( + convection = 0.5 * jnp.sum( self.derivative_operator[None, :] * u_outer_product_hat, axis=1, )