LSTM
功能
计算单层LSTM。
边界
【输入3-8】
X:,类型:tensor(float16), tensor(float), tensor(double)
W:,类型:tensor(float16), tensor(float), tensor(double)
R:,类型:tensor(float16), tensor(float), tensor(double)
B:,类型:tensor(float16), tensor(float), tensor(double)
sequence_lens:,类型:tensor(int32)
initial_h:,类型:tensor(float16), tensor(float), tensor(double)
initial_c:,类型:tensor(float16), tensor(float), tensor(double)
p:,类型:tensor(float16), tensor(float), tensor(double)
【输出0-3】
Y:,类型:tensor(float16), tensor(float), tensor(double)
Y_h:,类型:tensor(float16), tensor(float), tensor(double)
Y_c:,类型:tensor(float16), tensor(float), tensor(double)
【属性】
direction: string,默认值:forward
hidden_size: int
支持的ONNX版本
Opset v8/v9/v10/v11/v12/v13
父主题: 支持ONNX算子清单