LSTM（Core ML）的处理序列

Posted 2023-03-27

【中文标题】LSTM（Core ML）的处理序列

【英文标题】：Process sequence by LSTM (Core ML)

【发布时间】：2018-03-14 13:26:28

【问题描述】：

我已经通过 coremltools 将带有 LSTM 的 Caffe 模型转换为 CoreML。现在我正在尝试执行它。但是，我找不到处理整个序列的方法

np.ndarray( (7, #sequence
             1, # batch
             120, 1, 1)) #items dims

因为我找不到仅设置隐藏状态 (LSTM_1_c_in) 和初始历史记录 (LSTM_1_h_in) 的首字母的方法以及 自动在处理序列的下一项时使用先前的状态/结果。

它通过手动重启方法 'predict' 工作，并从以前的输出中手动设置 LSTM_1_c_in 和 LSTM_1_h_in（模型 reutrns LSTM_1_h_out 和 LSTM_1_c_out）。

是否可以通过 1 次运行处理 整个 序列？

附：使用 Swift 的方式也是可以接受的。

【参考方案1】：

例如，LSTM 的 num_outputs 等于 3。

如果 lstm 获得初始历史记录并声明为 np.ndarray((1,1,3))，它将只处理序列的一个元素。 这是我的错误

1234563 /li>