def test_entropy_1d(): diag = [1.0, 1.0] cov = tf.constant(diag) _assert_eq(multivariate_normal.entropy(cov=cov), stats.multivariate_normal.entropy(cov=np.diag(diag))) _assert_eq(multivariate_normal.entropy(cov=np.diag(diag)), stats.multivariate_normal.entropy(cov=np.diag(diag)))
def test_entropy_2d_full(): cm = [[1.0, 0.9], [0.9, 1.0]] cov = tf.constant(cm) _assert_eq(multivariate_normal.entropy(cov=cov), stats.multivariate_normal.entropy(cov=np.array(cm))) _assert_eq(multivariate_normal.entropy(cov=np.array(cm)), stats.multivariate_normal.entropy(cov=np.array(cm)))
def test_2d_full(self): cm = [[1.0, 0.9], [0.9, 1.0]] cov = tf.constant(cm) with self.test_session(): self.assertAllClose(multivariate_normal.entropy(cov=cov).eval(), stats.multivariate_normal.entropy(cov=np.array(cm))) self.assertAllClose(multivariate_normal.entropy(cov=np.array(cm)).eval(), stats.multivariate_normal.entropy(cov=np.array(cm)))
def test_1d(self): diag = [1.0, 1.0] cov = tf.constant(diag) with self.test_session(): self.assertAllClose(multivariate_normal.entropy(cov=cov).eval(), stats.multivariate_normal.entropy(cov=np.diag(diag))) self.assertAllClose(multivariate_normal.entropy(cov=np.diag(diag)).eval(), stats.multivariate_normal.entropy(cov=np.diag(diag)))
def test_entropy_empty(): _assert_eq(multivariate_normal.entropy(), stats.multivariate_normal.entropy())
def test_empty(self): with self.test_session(): self.assertAllClose(multivariate_normal.entropy().eval(), stats.multivariate_normal.entropy())