Category: Conditions
Short description: Select returns a tensor filled with the elements from the second or the third inputs, depending on the condition (the first input) value.
Detailed description
Select takes elements from then
input tensor or the else
input tensor based on a condition mask provided in the first input cond
. Before performing selection, input tensors then
and else
are broadcasted to each other if their shapes are different and auto_broadcast
attributes is not none
. Then the cond
tensor is one-way broadcasted to the resulting shape of broadcasted then
and else
. Broadcasting is performed according to auto_broadcast
value.
Attributes
Inputs:
cond
tensor with selection mask of type boolean
. The tensor can be 0D.then
the tensor with elements to take where the corresponding element in cond
is true. Arbitrary type that should match type of else
input tensor.else
the tensor with elements to take where the corresponding element in cond
is false. Arbitrary type that should match type of then
input tensor.Outputs:
then
and else
based on cond
and broadcasting rules. It has the same type of elements as then
and else
.Example