我已经通过coremltools将带有LSTM的Caffe模型转换为CoreML。现在我正在尝试执行它。但是,我找不到处理整个序列的方法
np.ndarray( (7, #sequence
1, # batch
120, 1, 1)) #items dims
因为我找不到设置隐藏状态( LSTM_1_c_in )和初始历史记录( LSTM_1_h_in )和自动的首字母的方法在处理下一个序列项时使用先前的状态/结果。
通过手动设置 LSTM_1_c_in 和 LSTM_1_h_in 从以前的输出中手动重启方法'预测',模型重新启动 LSTM_1_h_out 和分别为LSTM_1_c_out 。
是否可以通过1次运行处理整个序列?
P.S。使用Swift的方式也是可以接受的。
答案 0 :(得分:0)
例如,LSTM的num_outputs
等于3。
如果lstm获得初始历史记录并且状态为np.ndarray((1,1,3))
,它将只处理一个序列元素。 这是我的错误
如果lstm获得初始历史记录并且状态为np.ndarray((3))
,它将处理整个序列(但它只返回处理最后一个元素而没有完整历史记录的结果)。