Class ov::op::v1::OneHot#

class OneHot : public ov::op::Op#

OneHot operation.

Public Functions

OneHot() = default#

Constructs a one-hot operation.

OneHot(const Output<Node> &indices, const Output<Node> &depth, const Output<Node> &on_value, const Output<Node> &off_value, int64_t axis)#

Constructs a one-hot operation.

Parameters:

  • indices – Input tensor containing indices.

  • depth – Specifies number of classes and the size of one-hot dimension.

  • on_value – Specifies value that the locations in output tensor represented by indices in input take.

  • off_value – Specifies value that the locations in output tensor not represented by indices in input take.

  • axis – Axis along which one-hot representation in added.

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.

virtual bool has_evaluate() const override#

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

inline const int64_t &get_axis() const#
Returns:

The index of the one-hot axis.