tf.train.TrackableView

Собирает и сериализует представление отслеживаемого объекта.

tf.train.TrackableView(
    root
)

Пример использования:

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)

Передайте корень в tf.train.TrackableView.children(), чтобы получить словарь всех дочерних элементов, напрямую связанных с корнем по имени.

>>> 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`	Объект `Trackable`, переменные которого (включая переменные зависимостей, рекурсивно) должны быть сохранены. Может быть слабой ссылкой.

Атрибуты
`root`

Методы

`children`

Просмотреть исходный код

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

Возвращает все дочерние отслеживаемые объекты, присоединенные к объекту.

Аргументы
`obj`	Объект `Trackable`.
`save_type`	Строка, может быть 'savedmodel' или 'checkpoint'.
`**kwargs`	Параметры для получения дочерних элементов объекта.

Возвращает
Словарь всех дочерних элементов, присоединенных к объекту, с именем в качестве ключа и отслеживаемым объектом в качестве значения.

`descendants`

Просмотреть исходный код

descendants()

Возвращает список всех узлов из self.root, используя обход в ширину.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/train/TrackableView