code: def testNormalLogPDF(self):
      batch_size = 6
      mu = constant_op.constant([3.0] * batch_size)
      sigma = constant_op.constant([math.sqrt(10.0)] * batch_size)
      x = np.array([-2.5, 2.5, 4.0, 0.0, -1.0, 2.0], dtype=np.float32)
      normal = normal_lib.Normal(loc=mu, scale=sigma)
  
      log_pdf = normal.log_prob(x)
      self.assertAllEqual(
          self.evaluate(normal.batch_shape_tensor()),
          log_pdf.get_shape(),
      )
      self.assertAllEqual(
          self.evaluate(normal.batch_shape_tensor()),
          self.evaluate(log_pdf).shape,
      )
      self.assertAllEqual(normal.batch_shape, log_pdf.get_shape())
      self.assertAllEqual(
          normal.batch_shape, self.evaluate(log_pdf).shape
      )
  
      pdf = normal.prob(x)
      self.assertAllEqual(
          self.evaluate(normal.batch_shape_tensor()), pdf.get_shape()
      )
      self.assertAllEqual(
          self.evaluate(normal.batch_shape_tensor()),
          self.evaluate(pdf).shape,
      )
      self.assertAllEqual(normal.batch_shape, pdf.get_shape())
      self.assertAllEqual(normal.batch_shape, self.evaluate(pdf).shape)
  
      if not stats:
          return
      expected_log_pdf = stats.norm(
          self.evaluate(mu), self.evaluate(sigma)
      ).logpdf(x)
      self.assertAllClose(expected_log_pdf, self.evaluate(log_pdf))
      self.assertAllClose(np.exp(expected_log_pdf), self.evaluate(pdf))
  
imports: import importlib
  import math
  import numpy
  from tensorflow.python.eager import backprop
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import tensor_shape
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import gradients_impl
  from tensorflow.python.ops import nn_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.ops.distributions import kullback_leibler
  from tensorflow.python.ops.distributions import normal
  from tensorflow.python.platform import test
  from tensorflow.python.platform import tf_logging
label_desc: Use image node_id 9 for calling the NormalTest obj's underlying member method code with example usage: obj.testNormalLogPDF() without return types