The input condition. Must be of dtype bool.
If condition
is rank 1, a
may have a higher rank but
its first dimension must match the size of condition
.
A tensor with the same dtype as a
and with shape that is
compatible with a
.
A tensor with same dtype as a
and b
, and shape that is
broadcastable from a
and b
.
Generated using TypeDoc
Returns the elements, either
a
orb
depending on thecondition
.If the condition is true, select from
a
, otherwise select fromb
.