def testZipfMode(self):
    power_v = [10.0, 3.0, 2.5, 3.2, 1.1, 0.05]
    zipf = tfd.Zipf(power=power_v)
    self.assertEqual((6,), zipf.mode().shape)
    self.assertAllClose(self.evaluate(zipf.mode()), np.ones_like(power_v))
def testZipfMode(self):
    power_v = [10.0, 3.0, 2.5, 3.2, 1.1, 0.05]
    zipf = tfd.Zipf(power=power_v)
    self.assertEqual((6,), zipf.mode().shape)
    self.assertAllClose(self.evaluate(zipf.mode()), np.ones_like(power_v))