def testZipfStd(self):
    power_v = [4.0, 3.5, 4.5]
    zipf = tfd.Zipf(power=power_v, validate_args=True)
    self.assertEqual((3,), zipf.stddev().shape)
    stat_stddevs = np.vectorize(stats.zipf.std)(power_v)
    self.assertAllClose(self.evaluate(zipf.stddev()), stat_stddevs)
def testZipfStd(self):
    power_v = [4.0, 3.5, 4.5]
    zipf = tfd.Zipf(power=power_v, validate_args=True)
    self.assertEqual((3,), zipf.stddev().shape)
    stat_stddevs = np.vectorize(stats.zipf.std)(power_v)
    self.assertAllClose(self.evaluate(zipf.stddev()), stat_stddevs)