diff options
| author |  Joshua V. Dillon <jvdillon@google.com> | 2018-05-30 14:52:57 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-30 14:55:47 -0700 |
| commit | 176754d6cce54a971c98096f55251870708eea3e (patch) | |
| tree | 381b30f062f350bac3c5a6da9bd7a6d299c124a5 /tensorflow/contrib/distributions | |
| parent | 5c751fe8d766d4875cc99d58a536a29652685e26 (diff) | |
Add `fill_triangular_inverse`, which flattens a triangular matrix in a way such that:
# Lower triangular matrix
x = tf.matrix_band_part(x, -1, 0)
x == fill_triangular(fill_triangular_inverse(x))
Code by srvasude@ which I'm submitting on his behalf.
PiperOrigin-RevId: 198623887
Diffstat (limited to 'tensorflow/contrib/distributions')
| -rw-r--r-- | tensorflow/contrib/distributions/__init__.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/distributions/__init__.py b/tensorflow/contrib/distributions/__init__.py index ddf59891e6..802538ba97 100644 --- a/tensorflow/contrib/distributions/__init__.py +++ b/tensorflow/contrib/distributions/__init__.py @@ -32,6 +32,7 @@ from tensorflow.contrib.distributions.python.ops.conditional_distribution import from tensorflow.contrib.distributions.python.ops.conditional_transformed_distribution import * from tensorflow.contrib.distributions.python.ops.deterministic import * from tensorflow.contrib.distributions.python.ops.distribution_util import fill_triangular +from tensorflow.contrib.distributions.python.ops.distribution_util import fill_triangular_inverse from tensorflow.contrib.distributions.python.ops.distribution_util import matrix_diag_transform from tensorflow.contrib.distributions.python.ops.distribution_util import reduce_weighted_logsumexp from tensorflow.contrib.distributions.python.ops.distribution_util import softplus_inverse @@ -156,6 +157,7 @@ _allowed_symbols = [ 'kl_divergence', 'RegisterKL', 'fill_triangular', + 'fill_triangular_inverse', 'matrix_diag_transform', 'reduce_weighted_logsumexp', 'softplus_inverse', |