使用图表时,我们可以通过Session
的参数记录张量流操作的设备位置:
tf.Session(config=tf.ConfigProto(log_device_placement=True))
但是,在急切模式下,没有Session
。当急切执行时,我们怎么知道操作的设备放置?
答案 0 :(得分:0)
目前尚未在文档中明确宣传,但在热切模式下,每个Tensor
和Variable
都有一个device
字段,显示变量所在的位置或操作所在的位置的地方。
但是,AFAIK没有自动设备放置日志,无论何时运行操作都会触发。因此,在对象层次结构中进行操作的设备放置可能需要一些工作。