tf.train.TrackableView

Gathers and serializes a trackable view.

tf.train.TrackableView(
    root
)

Example usage:

class SimpleModule(tf.Module):
  def __init__(self, name=None):
    super().__init__(name=name)
    self.a_var = tf.Variable(5.0)
    self.b_var = tf.Variable(4.0)
    self.vars = [tf.Variable(1.0), tf.Variable(2.0)]
 
root = SimpleModule(name="root")
root.leaf = SimpleModule(name="leaf")
trackable_view = tf.train.TrackableView(root)

Pass root to tf.train.TrackableView.children() to get the dictionary of all children directly linked to root by name.

>>> trackable_view_children = trackable_view.children(root)
>>> for item in trackable_view_children.items():
...   print(item)
('a_var', <tf.Variable 'Variable:0' shape=() dtype=float32, numpy=5.0>)
('b_var', <tf.Variable 'Variable:0' shape=() dtype=float32, numpy=4.0>)
('vars', ListWrapper([<tf.Variable 'Variable:0' shape=() dtype=float32,
numpy=1.0>, <tf.Variable 'Variable:0' shape=() dtype=float32, numpy=2.0>]))
('leaf', ...)

root A Trackable object whose variables (including the variables of dependencies, recursively) should be saved. May be a weak reference.

root

Methods

children

View source

@classmethod
children(
    obj, save_type=base.SaveType.CHECKPOINT, **kwargs
)

Returns all child trackables attached to obj.

Args

obj A Trackable object. save_type A string, can be 'savedmodel' or 'checkpoint'. **kwargs kwargs to use when retrieving the object's children.

Returns
Dictionary of all children attached to the object with name to trackable.

descendants

View source

descendants()

Returns a list of all nodes from self.root using a breadth first traversal.