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
父主题: 支持ONNX算子清单