From 26840bf6a3472ca5bad095e113a461e8c4613016 Mon Sep 17 00:00:00 2001 From: jiangjinsheng Date: Tue, 19 May 2020 17:13:48 +0800 Subject: [PATCH] fixed comment for nn.lstm --- mindspore/nn/layer/lstm.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/mindspore/nn/layer/lstm.py b/mindspore/nn/layer/lstm.py index 06ad5154ab..86d0e8e44a 100755 --- a/mindspore/nn/layer/lstm.py +++ b/mindspore/nn/layer/lstm.py @@ -63,7 +63,7 @@ class LSTM(Cell): num_layers (int): Number of layers of stacked LSTM . Default: 1. has_bias (bool): Specifies whether has bias `b_ih` and `b_hh`. Default: True. batch_first (bool): Specifies whether the first dimension of input is batch_size. Default: False. - dropout (float): If not 0, append `Dropout` layer on the outputs of each + dropout (float, int): If not 0, append `Dropout` layer on the outputs of each LSTM layer except the last layer. Default 0. The range of dropout is [0.0, 1.0]. bidirectional (bool): Specifies whether this is a bidirectional LSTM. If set True, number of directions will be 2 otherwise number of directions is 1. Default: False.