Class ov::op::v1::Select

class Select : public ov::op::Op

Elementwise selection operation.

Inputs

Type

Description

arg0

\(\texttt{bool}[d_1,\dots,d_n]~(n \geq 0)\)

A tensor of any shape, with element bool.

arg1

\(E[d_1,\dots,d_n]~(n \geq 0)\)

A tensor of a shape that is broadcast-compatible with arg0, with any element type.

arg2

\(E[d_1,\dots,d_n]~(n \geq 0)\)

A tensor of a shape that is broadcast-compatible with arg0, and same element type as arg1.

auto_broadcast

AutoBroadcastSpec

Auto broadcast specification.

Type

Description

\(E[d_1,\dots,d_n]\)

The tensor \(T\), where \(T[i_1,\dots,i_n] = \texttt{arg1}[i_1,\dots,i_n]\text{ if }\texttt{arg0}[i_1,\dots,i_n] \neq 0\text{, else }\texttt{arg2}[i_1,\dots,i_n]\)

Public Functions

inline Select()

Constructs a selection operation.

Select(const Output<Node> &arg0, const Output<Node> &arg1, const Output<Node> &arg2, const AutoBroadcastSpec &auto_broadcast = AutoBroadcastSpec(AutoBroadcastType::NUMPY))

Constructs a selection operation.

Parameters
  • arg0Node that produces the first input tensor.

  • arg1Node that produces the second input tensor.

  • arg2Node that produces the third input tensor.

  • auto_broadcast – Auto broadcast specification. Default is Numpy-style implicit broadcasting.

virtual void validate_and_infer_types() override

Verifies that attributes and inputs are consistent and computes output shapes and element types. Must be implemented by concrete child classes so that it can be run any number of times.

Throws if the node is invalid.

inline virtual const AutoBroadcastSpec &get_autob() const override
Returns

the autobroadcasr spec

virtual bool has_evaluate() const override

Allows to get information about availability of evaluate method for the current operation.