标签： swift tensorflow lstm swift-for-tensorflow
我有一个带lstm层的简单tensorflow模型。我想将模型转换为.mlmodel格式。但是，我认为，到目前为止，CoreML不支持LSTM层，因此我需要快速编写自定义的lstm层。 如何编写该自定义图层？
答案 0 :(得分：0)
为什么不尝试使用Keras并使用不同的tensorflow或其他后端？ Keras具有最简单的界面之一。
我建议使用keras阅读张量流模型。然后使用下面的链接将keras转换为core-ml。
尝试使其简单。不复杂
https://heartbeat.fritz.ai/using-coremltools-to-convert-a-keras-model-to-core-ml-for-ios-d4a0894d4aba?gi=76f8b08071e9