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"