在TensorFlow中定义自定义模型/体系结构

Aut*_*Eng 2 deep-learning tensorflow

从我用TensorFlow玩的小东西,我看到它已经实现了像RNN/LSTM单元,ConvNets等架构.有没有办法定义一个"自定义"架构(例如一个带有少量卷积的"增强型"LSTM网络)层)?

Sun*_*Kim 5

是的,这完全有可能.LSTM或任何网络的输出是驾驶员用作另一网络的输入的张量.

请参阅https://github.com/jazzsaxmafia/show_and_tell.tensorflow,了解如何将它们组合在一起.

您可以在https://github.com/TensorFlowKR/awesome_tensorflow_implementations找到更多示例.