public final class
RefSwitch
Forwards the ref tensor `data` to the output port determined by `pred`.
If `pred` is true, the `data` input is forwarded to `output_true`. Otherwise, the data goes to `output_false`.
See also `Switch` and `Merge`.
Constants
String | OP_NAME | The name of this op, as known by TensorFlow core engine |
Public Methods
static <T extends TType> RefSwitch<T> | |
Output<T> |
outputFalse()
If `pred` is false, data will be forwarded to this output.
|
Output<T> |
outputTrue()
If `pred` is true, data will be forwarded to this output.
|
Inherited Methods
Constants
public static final String OP_NAME
The name of this op, as known by TensorFlow core engine
Constant Value:
"RefSwitch"
Public Methods
public static RefSwitch<T> create (Scope scope, Operand<T> data, Operand<TBool> pred)
Factory method to create a class wrapping a new RefSwitch operation.
Parameters
scope | current scope |
---|---|
data | The ref tensor to be forwarded to the appropriate output. |
pred | A scalar that specifies which output port will receive data. |
Returns
- a new instance of RefSwitch