RefSwitch

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>
create ( Scope scope, Operand <T> data, Operand < TBool > pred)
Factory method to create a class wrapping a new RefSwitch operation.
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

public Output <T> outputFalse ()

If `pred` is false, data will be forwarded to this output.

public Output <T> outputTrue ()

If `pred` is true, data will be forwarded to this output.