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

时间:2016-04-21 18:51:52

标签: tensorflow deep-learning

从我用TensorFlow玩的小东西中我发现它已经实现了像RNN / LSTM单元,ConvNets等架构。有没有办法定义一个人的“自定义”架构(例如“增强型”LSTM网络一些卷积层?)

1 个答案:

答案 0 :(得分:5)

是的,这完全有可能。 LSTM或任何网络的输出都是张量,其中cab用作另一个网络的输入。

了解如何在https://github.com/jazzsaxmafia/show_and_tell.tensorflow合并它们。

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