在使用图形时,我们可以通过Session
参数记录tensorflow操作的设备位置
tf.Session(config=tf.ConfigProto(log_device_placement=True))
但是,在急切模式下,没有Session
。当我们急切地执行操作时,我们如何知道操作的设备位置?
发布于 2018-06-17 05:37:52
它还没有明确地在文档中公布,但在急切模式下,每个Tensor
和Variable
都有一个device
字段,该字段显示变量的位置或操作发生的位置。
然而,AFAIK没有在运行操作时触发的自动设备放置日志。因此,可能需要一些工作才能获得发生在对象层次结构中的操作的设备放置。
https://stackoverflow.com/questions/50891957
复制相似问题