标签: c++ tensorflow
是否可以从session.Run提取(fetch_outputs)中获取指向GPU内存的指针?
session.Run
fetch_outputs
还可以从GPU馈送数据({{1)} / feed_dict吗?
我想避免大多数CPU-> GPU和GPU-> CPU传输(用于复制LSTM隐藏状态等)。
这个问题主要与C ++ API有关,尽管我也对Python API感兴趣。