diff --git a/mindspore/ops/operations/nn_ops.py b/mindspore/ops/operations/nn_ops.py index 3b8dc42ee0..ef83cbcf89 100644 --- a/mindspore/ops/operations/nn_ops.py +++ b/mindspore/ops/operations/nn_ops.py @@ -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)