Python tf.name_scope用法及代码示例

定义 Python 操作时使用的上下文管理器。

用法

tf.name_scope(
    name
)

这个上下文管理器推送一个名称范围，这将使添加在其中的所有操作的名称都有一个前缀。

例如，要定义一个名为 my_op 的新 Python 操作：

def my_op(a, b, c, name=None):
  with tf.name_scope("MyOp") as scope:
    a = tf.convert_to_tensor(a, name="a")
    b = tf.convert_to_tensor(b, name="b")
    c = tf.convert_to_tensor(c, name="c")
    # Define some computation that uses `a`, `b`, and `c`.
    return foo_op(..., name=scope)

执行时，张量 a , b , c 将具有名称 MyOp/a , MyOp/b 和 MyOp/c 。

在 tf.function 中，如果范围名称已存在，则通过附加 _n 使名称唯一。例如，第二次调用 my_op 将生成 MyOp_1/a 等。