swift - 如何为 tensorflow 快速编写 RNN/LSTM 自定义层？

Question

我有一个带有 lstm 层的简单张量流模型。我想将模型转换为 .mlmodel 格式。但是，我认为，到目前为止，CoreML 不支持 LSTM 层，因此我需要快速编写一个自定义的 lstm 层。
如何编写该自定义层？

score 0 · Accepted Answer

为什么不尝试使用 Keras 并使用不同的 tensorflow 或其他后端？Keras 拥有最简单的界面之一。

我建议使用 keras 阅读 tensorflow 模型。然后使用下面的链接将 keras 转换为 core-ml。

试着让它变得简单。不复杂

1 回答 1