Python tf.Module.with_name_scope用法及代码示例

用法

@classmethod
with_name_scope(
    method
)

装饰器自动进入模块名称范围。

class MyModule(tf.Module):
  @tf.Module.with_name_scope
  def __call__(self, x):
    if not hasattr(self, 'w'):
      self.w = tf.Variable(tf.random.normal([x.shape[1], 3]))
    return tf.matmul(x, self.w)

使用上述模块将产生 tf.Variable 和 tf.Tensor ，其名称包括模块名称：

mod = MyModule()
mod(tf.ones([1, 2]))
<tf.Tensor:shape=(1, 3), dtype=float32, numpy=..., dtype=float32)>
mod.w
<tf.Variable 'my_module/Variable:0' shape=(2, 3) dtype=float32,
numpy=..., dtype=float32)>