Как написать пользовательский слой RNN / LSTM в swift для тензорного потока?

Question

У меня есть простая модель тензорного потока со слоями lstm.Я хочу преобразовать модель в формат .mlmodel.Тем не менее, я думаю, что в настоящее время CoreML не поддерживает слои LSTM, и поэтому мне нужно написать собственный слой lstm в swift.
Как мне написать этот пользовательский слой?

Answer 1

Почему бы не попробовать Keras и использовать другой тензор потока или другие фоны?Keras имеет один из самых простых интерфейсов.

Я предлагаю прочитать модель тензорного потока с использованием keras.Затем используйте ссылку ниже, чтобы преобразовать кераты в core-ml.

Попробуйте сделать это проще.не сложно

https://heartbeat.fritz.ai/using-coremltools-to-convert-a-keras-model-to-core-ml-for-ios-d4a0894d4aba?gi=76f8b08071e9