tff.utils.assign

Creates an op that assigns target from 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.

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

A single op that represents the assignment.

TypeError If types mismatch.