如何在急切模式下放置设备?

时间:2018-06-16 21:37:52

标签: python tensorflow

使用图表时,我们可以通过Session的参数记录张量流操作的设备位置:

tf.Session(config=tf.ConfigProto(log_device_placement=True))

但是,在急切模式下,没有Session。当急切执行时,我们怎么知道操作的设备放置?

1 个答案:

答案 0 :(得分:0)

目前尚未在文档中明确宣传,但在热切模式下,每个TensorVariable都有一个device字段,显示变量所在的位置或操作所在的位置的地方。

但是,AFAIK没有自动设备放置日志,无论何时运行操作都会触发。因此,在对象层次结构中进行操作的设备放置可能需要一些工作。