下载
中文
注册

LSTM

功能

计算单层LSTM。

输入

3-8个输入:

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。

输出

三个输出:

Y:tensor,数据类型:float16。

Y_h:tensor,数据类型:float16。

Y_c:tensor,数据类型:float16。

属性

direction: string,默认值:forward。

hidden_size: int。

activation_alpha:(可选)某些激活函数使用的缩放值。

activation_beta:(可选)某些激活函数使用的缩放值。

clip:将张量的元素裁剪到 [-clip, +clip] 范围内,并应用于激活的输入。默认不做裁剪。

input_forget:默认为0;为1则耦合输入与遗忘门。

支持的ONNX版本

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