def testZipfMean(self):
    power_v = [2.0, 3.0, 2.5]
    zipf = tfd.Zipf(power=power_v)
    self.assertEqual((3,), zipf.mean().shape)
    self.assertAllClose(self.evaluate(zipf.mean()), stats.zipf.mean(power_v))
def testZipfMean(self):
    power_v = [2.0, 3.0, 2.5]
    zipf = tfd.Zipf(power=power_v)
    self.assertEqual((3,), zipf.mean().shape)
    self.assertAllClose(self.evaluate(zipf.mean()), stats.zipf.mean(power_v))