def testZipfShape(self):
    power = tf.constant(([3.0] * 5))
    zipf = tfd.Zipf(power=power, validate_args=True)
    self.assertEqual(self.evaluate(zipf.batch_shape_tensor()), (5,))
    self.assertEqual(zipf.batch_shape, tf.TensorShape([5]))
    self.assertAllEqual(self.evaluate(zipf.event_shape_tensor()), [])
    self.assertEqual(zipf.event_shape, tf.TensorShape([]))
def testZipfShape(self):
    power = tf.constant(([3.0] * 5))
    zipf = tfd.Zipf(power=power, validate_args=True)
    self.assertEqual(self.evaluate(zipf.batch_shape_tensor()), (5,))
    self.assertEqual(zipf.batch_shape, tf.TensorShape([5]))
    self.assertAllEqual(self.evaluate(zipf.event_shape_tensor()), [])
    self.assertEqual(zipf.event_shape, tf.TensorShape([]))