def _test(a, b): val_true = stats.beta.entropy(a, b) _assert_eq(beta.entropy(a, b), val_true) _assert_eq(beta.entropy(tf.constant(a), tf.constant(b)), val_true) _assert_eq(beta.entropy(tf.constant([a]), tf.constant(b)), val_true) _assert_eq(beta.entropy(tf.constant(a), tf.constant([b])), val_true) _assert_eq(beta.entropy(tf.constant([a]), tf.constant([b])), val_true)
def _test(self, a, b): val_true = stats.beta.entropy(a, b) self.assertAllClose(beta.entropy(a, b).eval(), val_true, atol=1e-4) self.assertAllClose(beta.entropy(tf.constant(a), tf.constant(b)).eval(), val_true, atol=1e-4) self.assertAllClose(beta.entropy(tf.constant([a]), tf.constant(b)).eval(), val_true, atol=1e-4) self.assertAllClose(beta.entropy(tf.constant(a), tf.constant([b])).eval(), val_true, atol=1e-4) self.assertAllClose(beta.entropy(tf.constant([a]), tf.constant([b])).eval(), val_true, atol=1e-4)
def entropy(self): return tf.reduce_sum(beta.entropy(self.a, self.b))
def entropy(self): return tf.reduce_sum(beta.entropy(self.alpha, self.beta))
def _test(self, a, b): val_true = stats.beta.entropy(a, b) self.assertAllClose(beta.entropy(a=a, b=b).eval(), val_true, atol=1e-4)