Join us at TensorFlow World, Oct 28-31. Use code TF20 for 20% off select passes. Register now

tff.utils.assign

View source

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.