TensorFlow 2.0 Beta is available Learn more

tff.utils.assign

Creates an op that assigns target from source.

tff.utils.assign(
    target,
    source
)

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

This utility function provides the exact same behavior as tf.Variable.assign, but it generalizes to a wider class of objects, including ordinary variables as well as various types of nested structures.

Args:

  • target: A nested structure composed of variables embedded in containers that are compatible with tf.nest, or instances of anonymous_tuple.AnonymousTuple.
  • source: A nsested structure composed of tensors, matching that of target.

Returns:

A single op that represents the assignment.

Raises:

  • TypeError: If types mismatch.