Change log_det_jacobian functions for SinhArcsinh to avoid NaN for large positive values.
Currently log1p(x ** 2) is computed which has a potential of returning an inf. This CL changes that computation for large x to remove that risk, maintaining accuracy. PiperOrigin-RevId: 163877450
Loading
Please sign in to comment