code: def test_get_masked_scores(self):
      scores = np.array(
          [[2.0, 5.0, -1e-3], [-1073.0, 4.2, -4.0]]
      ).astype(np.float32)
      mask = scores < 3.0
      target_res = np.array(
          [[2.0, -np.inf, -1e-3], [-1073.0, -np.inf, -4.0]]
      ).astype(np.float32)
      func_res = mi._get_masked_scores(scores, mask)
      self.assertAllEqual(self.evaluate(func_res), target_res)
  
file_path: probability/tensorflow_probability/python/vi/mutual_information_test.py
imports: import numpy
  import scipy
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.internal import test_util
  from tensorflow_probability.python.vi import mutual_information
label_desc: Use image node_id 3 for calling the MutualInformationTest obj's underlying member method code with example usage: obj.test_get_masked_scores() without return types