org.tensorflow.op.core

Class Switch<T>

java.lang.Object

org.tensorflow.op.PrimitiveOp

org.tensorflow.op.core.Switch<T>

Type Parameters:

T - data type for outputFalse() output

All Implemented Interfaces:

Op

public final class Switch<T>
extends PrimitiveOp

Forwards `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 `RefSwitch` and `Merge`.

Field Summary

Fields inherited from class org.tensorflow.op.PrimitiveOp

operation

Method Summary

Modifier and Type	Method and Description
static <T> Switch<T>	create(Scope scope, Operand<T> data, Operand<Boolean> pred) Factory method to create a class to wrap a new Switch operation to the graph.
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.

Methods inherited from class org.tensorflow.op.PrimitiveOp

equals, hashCode, toString

Methods inherited from class java.lang.Object

clone, finalize, getClass, notify, notifyAll, wait, wait, wait

Method Detail

create

public static <T> Switch<T> create(Scope scope,
                                   Operand<T> data,
                                   Operand<Boolean> pred)

Factory method to create a class to wrap a new Switch operation to the graph.

Parameters:

scope - current graph scope

data - The tensor to be forwarded to the appropriate output.

pred - A scalar that specifies which output port will receive data.

Returns:

a new instance of Switch

outputFalse

public Output<T> outputFalse()

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

outputTrue

public Output<T> outputTrue()

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