我想从TensorFlow中定义为DNN的分布中进行蒙特卡洛(MC)采样。采样代码需要一个自定义函数,该函数很难在TensorFlow代码内部使用。
由于MC需要进行大量的DNN调用,因此我必须运行(并关闭)会话多次才能从DNN进行采样。分析表明,与session.run()
相关的开销是我代码中的当前瓶颈。
TensorFlow似乎有一个C++ API,但它也需要运行一个会话。
有人知道session.run()
的开销是否在TensorFlow的C ++版本中减少了?在花时间重新编写代码之前,很高兴知道这一点。
使用外部(python或c ++)函数进行MC采样的有效TensorFlow方法是什么[无法在TensorFlow中定义]?