def testExponentialEntropy(self):
    with session.Session():
      lam_v = np.array([1.0, 4.0, 2.5])
      exponential = exponential_lib.Exponential(rate=lam_v)
      self.assertEqual(exponential.entropy().get_shape(), (3,))
      if not stats:
        return
      expected_entropy = stats.expon.entropy(scale=1 / lam_v)
      self.assertAllClose(exponential.entropy().eval(), expected_entropy)