View source on GitHub |
A context manager for use when defining a Python op.
tf.keras.backend.name_scope(
name
)
This context manager pushes a name scope, which will make the name of all operations added within it have a prefix.
For example, to define a new Python op called my_op
:
def my_op(a):
with tf.name_scope("MyOp") as scope:
a = tf.convert_to_tensor(a, name="a")
# Define some computation that uses a
.
return foo_op(..., name=scope)
When executed, the Tensor a
will have the name MyOp/a
.
name
: The prefix to use on all names created within the name scope.Name scope context manager.