diff options
| author | anderspapitto <anderspapitto@gmail.com> | 2018-06-20 14:46:09 -0700 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-06-20 17:46:09 -0400 |
| commit | 544690bf4e2f49246e777040abeda560f097b715 (patch) | |
| tree | fc56193a6352418740832a3258b89b5b4c010cbf /torch | |
| parent | 48e90e3339b8b027e2525da6d1977204ecfbff73 (diff) | |
| download | pytorch-544690bf4e2f49246e777040abeda560f097b715.tar.gz pytorch-544690bf4e2f49246e777040abeda560f097b715.tar.bz2 pytorch-544690bf4e2f49246e777040abeda560f097b715.zip | |
Update rnn.py (#8705)
Diffstat (limited to 'torch')
| -rw-r--r-- | torch/nn/modules/rnn.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/torch/nn/modules/rnn.py b/torch/nn/modules/rnn.py index 9f1c095e90..4d60bb6331 100644 --- a/torch/nn/modules/rnn.py +++ b/torch/nn/modules/rnn.py @@ -248,7 +248,7 @@ class RNN(RNNBase): where :math:`h_t` is the hidden state at time `t`, :math:`x_t` is the input at time `t`, and :math:`h_{(t-1)}` is the hidden state of the previous layer at time `t-1` or the initial hidden state at time `0`. - If :attr:`nonlinearity`='relu', then `ReLU` is used instead of `tanh`. + If :attr:`nonlinearity` is `'relu'`, then `ReLU` is used instead of `tanh`. Args: input_size: The number of expected features in the input `x` @@ -544,7 +544,7 @@ class RNNCell(RNNCellBase): h' = \tanh(w_{ih} x + b_{ih} + w_{hh} h + b_{hh}) - If :attr:`nonlinearity`='relu', then ReLU is used in place of tanh. + If :attr:`nonlinearity` is `'relu'`, then ReLU is used in place of tanh. Args: input_size: The number of expected features in the input `x` |