---
title: LSTM
description: "计算单层LSTM。"
url: https://www.hiascend.com/document/detail/zh/canncommercial/latest/API/aolapi/atlasoxol_09_0090.html
sourcePath: /source/zh/canncommercial/900/API/aolapi/atlasoxol_09_0090.html
indexId: 561ae08c5a6b273713f4b2808c45b3a4a5a81f949bf783cb99161f1564dda47063
---
# LSTM

#### 功能

计算单层LSTM。


#### 输入

- x：输入Tensor，数据类型支持float16。
- w：输入Tensor，数据类型支持float16。
- r：输入Tensor，数据类型支持float16。
- b（可选）：输入Tensor，数据类型支持float16。
- sequence_lens（可选）：输入Tensor，数据类型支持int32。
- initial_h（可选）：输入Tensor，数据类型支持float16。
- initial_c（可选）：输入Tensor，数据类型支持float16。
- p（可选）：输入Tensor，数据类型支持float16。


#### 属性

- direction： 数据类型为string，默认值：forward。
- hidden_size： 数据类型为int。
- activation_alpha（可选）：数据类型为float，某些激活函数使用的缩放值。
- activation_beta（可选）：数据类型为float，某些激活函数使用的缩放值。
- clip：数据类型为float，将Tensor的元素裁剪到 [-clip， +clip] 范围内，并应用于激活的输入。默认不做裁剪。
- input_forget：数据类型为int32，默认为0；为1则耦合输入与遗忘门。


#### 输出

- y：输出Tensor，数据类型：float16。
- y_h：输出Tensor，数据类型：float16。
- y_c：输出Tensor，数据类型：float16。


#### 约束与限制

无。


#### 支持的ONNX版本

Opset v8/v9/v10/v11/v12/v13/v14/v15/v16