norm = tf.sqrt(2 * np.pi * sigma**2) * tf.math.erf(
        t_final / (np.sqrt(8) * sigma)
    ) - t_final * tf.exp(-(t_final**2) / (8 * sigma**2))