diff options
| author | Francisco Massa <fvsmassa@gmail.com> | 2017-06-19 17:21:04 -0400 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-06-19 18:18:01 -0400 |
| commit | 76ee014d10ea17897a5aa7edf641b9c00d6b7fd0 (patch) | |
| tree | 73e9e8fa98216dcef51c36e7799f44611d795c2d /docs | |
| parent | f619ac6ac9346d2ef974711637cc3b116a35e1f0 (diff) | |
| download | pytorch-76ee014d10ea17897a5aa7edf641b9c00d6b7fd0.tar.gz pytorch-76ee014d10ea17897a5aa7edf641b9c00d6b7fd0.tar.bz2 pytorch-76ee014d10ea17897a5aa7edf641b9c00d6b7fd0.zip | |
Add documentation to SELU and AlphaDropout
Diffstat (limited to 'docs')
| -rw-r--r-- | docs/source/nn.rst | 22 |
1 files changed, 22 insertions, 0 deletions
diff --git a/docs/source/nn.rst b/docs/source/nn.rst index 72fa10f234..ff6a80737f 100644 --- a/docs/source/nn.rst +++ b/docs/source/nn.rst @@ -230,6 +230,12 @@ Non-linear Activations .. autoclass:: ELU :members: +:hidden:`SELU` +~~~~~~~~~~~~~ + +.. autoclass:: SELU + :members: + :hidden:`PReLU` ~~~~~~~~~~~~~~~ @@ -424,6 +430,12 @@ Dropout layers .. autoclass:: Dropout3d :members: +:hidden:`AlphaDropout` +~~~~~~~~~~~~~~~~~~~ + +.. autoclass:: AlphaDropout + :members: + Sparse layers ---------------------------------- @@ -770,6 +782,11 @@ Non-linear activation functions .. autofunction:: elu +:hidden:`selu` +~~~~~~~~~~~~~ + +.. autofunction:: selu + :hidden:`leaky_relu` ~~~~~~~~~~~~~~~~~~~~ @@ -864,6 +881,11 @@ Dropout functions .. autofunction:: dropout +:hidden:`alpha_dropout` +~~~~~~~~~~~~~~~~~ + +.. autofunction:: alpha_dropout + Distance functions ---------------------------------- |