通过在Tensorflow中运行会话来最大程度地减少开销

时间:2019-03-28 17:25:10

标签: python c++ tensorflow session

我想从TensorFlow中定义为DNN的分布中进行蒙特卡洛(MC)采样。采样代码需要一个自定义函数,该函数很难在TensorFlow代码内部使用。

由于MC需要进行大量的DNN调用,因此我必须运行(并关闭)会话多次才能从DNN进行采样。分析表明,与session.run()相关的开销是我代码中的当前瓶颈。

TensorFlow似乎有一个C++ API,但它也需要运行一个会话。

  1. 有人知道session.run()的开销是否在TensorFlow的C ++版本中减少了?在花时间重新编写代码之前,很高兴知道这一点。

  2. 使用外部(python或c ++)函数进行MC采样的有效TensorFlow方法是什么[无法在TensorFlow中定义]?

0 个答案:

没有答案