Missed TensorFlow World? Check out the recap. Learn more

tff.utils.assign

View source on GitHub

Creates an op that assigns target from source.

tff.utils.assign(
    target,
    source
)

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.