Python tf.get_current_name_scope用法及代码示例

返回由 tf.name_scope(...) s 指定的当前全名范围。

用法

tf.get_current_name_scope()

例如，

with tf.name_scope("outer"):
  tf.get_current_name_scope()  # "outer"

  with tf.name_scope("inner"):
    tf.get_current_name_scope()  # "outer/inner"

换句话说，tf.get_current_name_scope() 返回将添加到前面的操作名称前缀，如果在该位置创建操作。

请注意，@tf.function 会重置名称范围堆栈，如下所示。

with tf.name_scope("outer"):

  @tf.function
  def foo(x):
    with tf.name_scope("inner"):
      return tf.add(x * x)  # Op name is "inner/Add", not "outer/inner/Add"