Browse Source

!15086 update document of BasicLSTMCell

From: @mind-lh
Reviewed-by: @liangchenghui,@liangchenghui
Signed-off-by: @liangchenghui,@liangchenghui
tags/v1.2.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
5904836ee8
1 changed files with 6 additions and 1 deletions
  1. +6
    -1
      mindspore/ops/operations/nn_ops.py

+ 6
- 1
mindspore/ops/operations/nn_ops.py View File

@@ -7128,6 +7128,9 @@ class CTCGreedyDecoder(PrimitiveWithCheck):

class BasicLSTMCell(PrimitiveWithInfer):
r"""
It's similar to operator DynamicRNN. BasicLSTMCell will be deprecated in the future.
Please use DynamicRNN instead.

Applies the long short-term memory (LSTM) to the input.

.. math::
@@ -7191,7 +7194,7 @@ class BasicLSTMCell(PrimitiveWithInfer):
TypeError: If dtype of `x`, `h`, `c` or `w` is neither float16 nor float32.

Supported Platforms:
``Ascend``
Deprecated

Examples:
>>> np.random.seed(0)
@@ -7213,6 +7216,8 @@ class BasicLSTMCell(PrimitiveWithInfer):
[[0.0000e+00, 0.0000e+00]]))
"""

# deprecate_new_name = "BasicLSTMCell"

@prim_attr_register
def __init__(self, keep_prob=1.0, forget_bias=1.0, state_is_tuple=True, activation='tanh'):
self.keep_prob = validator.check_value_type("keep_prob", keep_prob, [float], self.name)


Loading…
Cancel
Save