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
structure.Struct .
|
source
|
A nested structure composed of tensors, matching that of target .
|
Returns |
A single op that represents the assignment.
|
Raises |
TypeError
|
If types mismatch.
|