ngraph.opset5.select

ngraph.opset5.select(cond: Union[_pyngraph.Node, int, float, numpy.ndarray], then_node: Union[_pyngraph.Node, int, float, numpy.ndarray], else_node: Union[_pyngraph.Node, int, float, numpy.ndarray], auto_broadcast: str = 'numpy', name: Optional[str] = None) _pyngraph.Node

Perform an element-wise selection operation on input tensors.

Parameters
  • cond – Tensor with selection mask of type boolean.

  • then_node – Tensor providing data to be selected if respective cond item value is True.

  • else_node – Tensor providing data to be selected if respective cond item value is False.

  • auto_broadcast – Mode specifies rules used for auto-broadcasting of input tensors.

  • name – The optional new name for output node.

Returns

The new node with values selected according to provided arguments.