TensorFlow中的RNN实时实现

时间:2016-07-03 10:35:24

标签: python c++ machine-learning tensorflow

我想在TensorFlow中实现一个RNN来跟踪几个对象的位置。我想要一个实时实现,所以我会给网络一个数据点。我在TensorFlow中看到的示例输入始终是不同时间的数据点序列。如何在此时只输入一个数据点?

1 个答案:

答案 0 :(得分:0)

作为Tensorflow文档, https://www.tensorflow.org/api_docs/python/tf/nn/dynamic_rnn

  • 您可以将初始状态指定为RNN网络的输入,
  • 此外,Tensorflow RNN网络输出最终状态。

因此,您可以将输入作为[batchsize,1,inputsize]的形状(不要忘记初始状态;),并将接收具有以下形状的输出[batchsize,1,outputsize]和最终状态。

然后,您可以将上一次执行的最终状态作为初始状态输入到下一次执行。

这将允许您创建可以实时运行的RNN;)