shape = tf.reshape(
        tfp.math.interp_regular_1d_grid(
            t_interp,
            t_bin_start,
            t_bin_end,