Skip to content

Commit

Permalink
Add reference to tf.Module RFC.
Browse files Browse the repository at this point in the history
h/t. tensorflow/community#56

PiperOrigin-RevId: 250436499
Change-Id: I9332d90f6331756f12907566af9a9e23f86aded8
  • Loading branch information
tomhennigan authored and copybara-github committed May 29, 2019
1 parent 200151a commit 804f7d4
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions sonnet/src/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -297,6 +297,14 @@ class Module(six.with_metaclass(ModuleMetaclass, tf.Module)):
>>> mod = MultiplyModule()
>>> mod(1.)
<tf.Tensor: ... numpy=2.0>
Sonnet modules are a layer on top of `tf.Module`, implementing automatic name
scoping as described in the original RFC [0]_.
References:
.. [0] Hennigan, Tom, et al. "RFC: Stateful Containers with tf.Module."
TensorFlow Community RFCs, Google / DeepMind, 17 Jan. 2019,
https://github.com/tensorflow/community/pull/56
"""

def __init__(self, name: Optional[Text] = None):
Expand Down

0 comments on commit 804f7d4

Please sign in to comment.