diff options
author | Scott Zhu <scottzhu@google.com> | 2018-04-13 17:52:20 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-04-13 17:57:27 -0700 |
commit | 3652556dab3ebfe0152232facc7304fe5754aecb (patch) | |
tree | 9a9cecde4c85dc53548a185f9bd6d7c6e0591262 /tensorflow/contrib/sparsemax | |
parent | ef24ad14502e992716c49fdd5c63e6b2c2fb6b5a (diff) |
Merge changes from github.
PiperOrigin-RevId: 192850372
Diffstat (limited to 'tensorflow/contrib/sparsemax')
-rw-r--r-- | tensorflow/contrib/sparsemax/__init__.py | 2 | ||||
-rw-r--r-- | tensorflow/contrib/sparsemax/python/ops/sparsemax.py | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/sparsemax/__init__.py b/tensorflow/contrib/sparsemax/__init__.py index 19d213fb3e..7bc726f4a8 100644 --- a/tensorflow/contrib/sparsemax/__init__.py +++ b/tensorflow/contrib/sparsemax/__init__.py @@ -14,7 +14,7 @@ # ============================================================================== """Module that implements sparsemax and sparsemax loss, see [1]. -[1] https://arxiv.org/abs/1602.02068 +[1]: https://arxiv.org/abs/1602.02068 ## Sparsemax diff --git a/tensorflow/contrib/sparsemax/python/ops/sparsemax.py b/tensorflow/contrib/sparsemax/python/ops/sparsemax.py index 890ca20f4c..e617af2ff1 100644 --- a/tensorflow/contrib/sparsemax/python/ops/sparsemax.py +++ b/tensorflow/contrib/sparsemax/python/ops/sparsemax.py @@ -31,7 +31,7 @@ def sparsemax(logits, name=None): """Computes sparsemax activations [1]. For each batch `i` and class `j` we have - sparsemax[i, j] = max(logits[i, j] - tau(logits[i, :]), 0) + $$sparsemax[i, j] = max(logits[i, j] - tau(logits[i, :]), 0)$$ [1]: https://arxiv.org/abs/1602.02068 |