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