total_variation.jax¶
JAX implementation for Total Variation scores.
Functions¶
Computes the Total Variation score using JAX arrays. |
Back to top