TensorFlow 2.0 Beta is available Learn more

tff.utils.get_variables

Creates a set of variables that matches the given type_spec.

tff.utils.get_variables(
    name,
    type_spec,
    **kwargs
)

Defined in python/core/utils/tf_computation_utils.py.

Args:

  • name: The common name to use for the scope in which all of the variables are to be created.
  • type_spec: An instance of tff.Type or something convertible to it. The type signature may only be composed of tensor types and named tuples, possibly nested.
  • **kwargs: Additional keyword args to pass to tf.get_variable calls.

Returns:

Either a single variable when invoked with a tensor TFF type, or a nested structure of variables created in the appropriately-named variable scopes made up of anonymous tuples if invoked with a named tuple TFF type.

Raises:

  • TypeError: if type_spec is not a type signature composed of tensor and named tuple TFF types.