T
- data type for outputRef()
output@Operator public final class Assign<T> extends PrimitiveOp implements Operand<T>
This operation outputs "ref" after the assignment is done. This makes it easier to chain operations that need to use the reset value.
Modifier and Type | Class and Description |
---|---|
static class |
Assign.Options
Optional attributes for
Assign |
operation
Modifier and Type | Method and Description |
---|---|
Output<T> |
asOutput()
Returns the symbolic handle of a tensor.
|
static <T> Assign<T> |
create(Scope scope,
Operand<T> ref,
Operand<T> value,
Assign.Options... options)
Factory method to create a class wrapping a new Assign operation.
|
Output<T> |
outputRef()
= Same as "ref".
|
static Assign.Options |
useLocking(Boolean useLocking) |
static Assign.Options |
validateShape(Boolean validateShape) |
equals, hashCode, op, toString
public static <T> Assign<T> create(Scope scope, Operand<T> ref, Operand<T> value, Assign.Options... options)
scope
- current scoperef
- Should be from a `Variable` node. May be uninitialized.value
- The value to be assigned to the variable.options
- carries optional attributes valuespublic static Assign.Options validateShape(Boolean validateShape)
validateShape
- If true, the operation will validate that the shape
of 'value' matches the shape of the Tensor being assigned to. If false,
'ref' will take on the shape of 'value'.public static Assign.Options useLocking(Boolean useLocking)
useLocking
- If True, the assignment will be protected by a lock;
otherwise the behavior is undefined, but may exhibit less contention.public Output<T> outputRef()
public Output<T> asOutput()
Operand
Inputs to TensorFlow operations are outputs of another TensorFlow operation. This method is used to obtain a symbolic handle that represents the computation of the input.
asOutput
in interface Operand<T>
OperationBuilder.addInput(Output)
Copyright © 2022. All rights reserved.