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