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`.

Public Methods

static <T> RefSwitch<T>
create(Scope scope, Operand<T> data, Operand<Boolean> 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

org.tensorflow.op.PrimitiveOp
final boolean
equals(Object obj)
final int
Operation
op()
Returns the underlying Operation
final String
boolean
equals(Object arg0)
final Class<?>
getClass()
int
hashCode()
final void
notify()
final void
notifyAll()
String
toString()
final void
wait(long arg0, int arg1)
final void
wait(long arg0)
final void
wait()

Public Methods

public static RefSwitch<T> create (Scope scope, Operand<T> data, Operand<Boolean> 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.