def test_compute_lower_percentile_means(self): # Shape = [2, 3, 3]. x = tf.constant([[[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]], [[11.0, 12.0, 13.0], [14.0, 15.0, 16.0], [17.0, 18.0, 19.0]]]) lower_half = data_utils.compute_lower_percentile_means(x, axis=[-2, -1]) self.assertAllClose(lower_half, [3.0, 13.0])
def compute_lower_half_negative_log_mean(x): return -tf.math.log( data_utils.compute_lower_percentile_means( x, axis=[-2, -1], q=50))