openvino.runtime.opset3.one_hot#
- openvino.runtime.opset3.one_hot(indices: Node | int | float | ndarray, depth: Node | int | float | ndarray, on_value: Node | int | float | ndarray, off_value: Node | int | float | ndarray, axis: int, name: str | None = None) Node #
Create node performing one-hot encoding on input data.
- Parameters:
indices – Input tensor of rank N with indices of any supported integer data type.
depth – Scalar of any supported integer type that specifies number of classes and the size of one-hot dimension.
on_value – Scalar of any type that is the value that the locations in output tensor represented by indices in input take.
off_value – Scalar of any type that is the value that the locations not represented by indices in input take.
name – The optional name for new output node.
- Returns:
New node performing one-hot operation.