diff options
| author | Tongzhou Wang <SsnL@users.noreply.github.com> | 2018-04-17 22:06:54 -0400 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2018-04-17 22:06:54 -0400 |
| commit | 1c01eabd3cc8402e6de815245774278af3ad0c7f (patch) | |
| tree | c79d47544f4e90addb9165fe3ae377201b8d01d3 /docs | |
| parent | c43c911662600e3d0a780ce3d23bd9d853ded590 (diff) | |
| download | pytorch-1c01eabd3cc8402e6de815245774278af3ad0c7f.tar.gz pytorch-1c01eabd3cc8402e6de815245774278af3ad0c7f.tar.bz2 pytorch-1c01eabd3cc8402e6de815245774278af3ad0c7f.zip | |
Codemod to update our codebase to 0.4 standard (#6641)
* Codemod to update our codebase to 0.4 standard
* Update some of the test scri[ts
* remove Variable in test_clip_grad_value
* fix _symbolic_override_wrapper_maker
Diffstat (limited to 'docs')
| -rw-r--r-- | docs/source/nn.rst | 20 |
1 files changed, 14 insertions, 6 deletions
diff --git a/docs/source/nn.rst b/docs/source/nn.rst index 55d7f62d7f..acebfc5ecf 100644 --- a/docs/source/nn.rst +++ b/docs/source/nn.rst @@ -40,7 +40,7 @@ Containers .. autoclass:: ParameterList :members: -Convolution Layers +Convolution layers ---------------------------------- :hidden:`Conv1d` @@ -81,7 +81,7 @@ Convolution Layers :members: -Pooling Layers +Pooling layers ---------------------------------- :hidden:`MaxPool1d` @@ -193,7 +193,7 @@ Pooling Layers :members: -Padding Layers +Padding layers -------------- :hidden:`ReflectionPad1d` @@ -251,8 +251,8 @@ Padding Layers :members: -Non-linear Activations (weighed sum+nonlinearity) -------------------------------------------------- +Non-linear activations (weighted sum, nonlinearity) +--------------------------------------------------- :hidden:`ELU` ~~~~~~~~~~~~~ @@ -356,7 +356,7 @@ Non-linear Activations (weighed sum+nonlinearity) .. autoclass:: Threshold :members: -Non-linear Activations (Other) +Non-linear activations (other) ------------------------------ :hidden:`Softmin` @@ -1194,6 +1194,14 @@ Vision functions .. autofunction:: affine_grid +DataParallel functions (multi-GPU, distributed) +-------------------------------------------- + +:hidden:`data_parallel` +~~~~~~~~~~~~~~~~~~~~~~ + +.. autofunction:: torch.nn.parallel.data_parallel + torch.nn.init ============= |