mirror of
https://github.com/DifferentiableUniverseInitiative/JaxPM.git
synced 2025-04-07 20:30:54 +00:00
10 lines
No EOL
184 B
Python
10 lines
No EOL
184 B
Python
import jax.numpy as jnp
|
|
|
|
def MSE(x , y):
|
|
return jnp.mean((x - y)**2)
|
|
|
|
def MSE_3D(x , y):
|
|
return ((x - y)**2).mean(axis=0)
|
|
|
|
def MSRE(x , y):
|
|
return jnp.mean(((x - y)/ y)**2) |