session。在GPU上使用fetches和feed_dict运行

时间:2019-07-15 20:16:30

标签: c++ tensorflow

是否可以从session.Run提取(fetch_outputs)中获取指向GPU内存的指针?

还可以从GPU馈送数据({{1)} / feed_dict吗?

我想避免大多数CPU-> GPU和GPU-> CPU传输(用于复制LSTM隐藏状态等)。

这个问题主要与C ++ API有关,尽管我也对Python API感兴趣。

0 个答案:

没有答案