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 等。