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.
|
View source on GitHub