!29985 add note for GRU on Ascend

Merge pull request !29985 from 吕昱峰(Nate.River)/code_docs_master
This commit is contained in:
i-robot 2022-02-14 03:35:13 +00:00 committed by Gitee
commit f25db3edd9
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 0 deletions

View File

@ -683,6 +683,9 @@ class GRU(_RNNBase):
`Learning Phrase Representations using RNN EncoderDecoder for Statistical Machine Translation
<https://aclanthology.org/D14-1179.pdf>`_.
Note:
When using GRU on Ascend, the hidden size only supports multiples of 16.
Args:
input_size (int): Number of features of input.
hidden_size (int): Number of features of hidden layer.