Class name_scope
A context manager for use when defining a Python op. name_scopeInherits From:
Aliases:
- Class
tf.compat.v2.name_scope
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 calledmy_op
:
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)
When exec
uted, the Tensors a
, b
, c
, will ha
ve na
mes MyOp/a
, MyOp/b
, a
nd MyOp/c
.
If the scope name already exists, the name will be made unique by appending _n
. For example, calling my_op
the second time will generate MyOp_1/a
, etc.
init
__init__(name)
Initialize the context manager.
Args:
name
: The prefix to use on allname
s created within thename
scope.
Raises:
ValueError
: If name is None, or not a string.
Properties
name
Methods
enter
__enter__()
Start the scope block.
Returns:
The scope name.
Raises:
ValueError
: if neithername
nordefault_name
is provided butvalues
are.
exit
__exit__(
type_arg,
value_arg,
traceback_arg
)