Example #1
0
 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])
Example #2
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))