class_docstr: Well tested because this is a simple override of the base class.
class_docstr_tok: ["Well","tested","because","this","is","a","simple","override","of","the","base","class","."]
code: def testImplicitLargeDiag(self):
      mu = np.array(
          [[1.0, 2, 3], [11, 22, 33]]
      )  # shape: [b, k] = [2, 3]
      u = np.array(
          [
              [[1.0, 2], [3, 4], [5, 6]],
              [[0.5, 0.75], [1, 0.25], [1.5, 1.25]],
          ]
      )  # shape: [b, k, r] = [2, 3, 2]
      m = np.array([[0.1, 0.2], [0.4, 0.5]])  # shape: [b, r] = [2, 2]
      scale = np.stack(
          [
              np.eye(3)
              + np.matmul(
                  np.matmul(u[0], np.diag(m[0])), np.transpose(u[0])
              ),
              np.eye(3)
              + np.matmul(
                  np.matmul(u[1], np.diag(m[1])), np.transpose(u[1])
              ),
          ]
      )
      cov = np.stack(
          [
              np.matmul(scale[0], scale[0].T),
              np.matmul(scale[1], scale[1].T),
          ]
      )
      tf1.logging.vlog(2, "expected_cov:\n{}".format(cov))
      mvn = mvdpl.MultivariateNormalDiagPlusLowRank(
          loc=mu,
          scale_perturb_factor=u,
          scale_perturb_diag=m,
          validate_args=True,
      )
      self.assertAllClose(
          cov, self.evaluate(mvn.covariance()), atol=0.0, rtol=1e-6
      )
  
file_path: probability/tensorflow_probability/python/distributions/mvn_diag_plus_low_rank_test.py
imports: import numpy
  import tensorflow.compat.v1
  import tensorflow.compat.v2
  from tensorflow_probability.python.distributions import kullback_leibler
  from tensorflow_probability.python.distributions import mvn_diag
  from tensorflow_probability.python.distributions import mvn_diag_plus_low_rank
  from tensorflow_probability.python.distributions import mvn_tril
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 8 for calling the MultivariateNormalDiagPlusLowRankTest obj's underlying member method code with example usage: obj.testImplicitLargeDiag() without return types