offset = tf.exp(-(t_final**2) / (8 * sigma**2))