如何让设备成为Op或将被置于？

Question

此问题类似于：How to get current TensorFlow name scope

但是这个问题与设备范围有关。我理解它稍微复杂一点，因为设备可能没有明确作用域，也不能保证。

是否有任何方式来深入了解这一点？

Answer 1

我使用以下实用程序

class _DeviceCaptureOp(object):
  def __init__(self):
    self.device = None
  def _set_device(self, device):
    self.device = device

def get_current_device():
  """Returns device string of current graph context."""

  g = tf.get_default_graph()
  op = _DeviceCaptureOp()
  g._apply_device_functions(op)
  return op.device