summaryrefslogtreecommitdiff
path: root/docs/source
diff options
context:
space:
mode:
authorSoumith Chintala <soumith@gmail.com>2017-09-14 09:41:04 -0400
committerSoumith Chintala <soumith@gmail.com>2017-09-14 09:41:04 -0400
commitce4932f8a4223e923d33afb45b3e0d63aefa6fb2 (patch)
treebc7a28c38d16c16f5caedbea0e14fba03ac39880 /docs/source
parent80d229b0e70ba9c2be04bc04f03c1e48d826f273 (diff)
downloadpytorch-ce4932f8a4223e923d33afb45b3e0d63aefa6fb2.tar.gz
pytorch-ce4932f8a4223e923d33afb45b3e0d63aefa6fb2.tar.bz2
pytorch-ce4932f8a4223e923d33afb45b3e0d63aefa6fb2.zip
add softmax2d docs
Diffstat (limited to 'docs/source')
-rw-r--r--docs/source/nn.rst6
1 files changed, 6 insertions, 0 deletions
diff --git a/docs/source/nn.rst b/docs/source/nn.rst
index d358f6258e..b7d2507a5e 100644
--- a/docs/source/nn.rst
+++ b/docs/source/nn.rst
@@ -314,6 +314,12 @@ Non-linear Activations
.. autoclass:: Softmax
:members:
+:hidden:`Softmax2d`
+~~~~~~~~~~~~~~~~~~~
+
+.. autoclass:: Softmax2d
+ :members:
+
:hidden:`LogSoftmax`
~~~~~~~~~~~~~~~~~~~~