openvino.runtime.opset4.softmax#
- openvino.runtime.opset4.softmax(data: Node | int | float | ndarray, axis: int, name: str | None = None) Node #
Apply softmax operation on each element of input tensor.
- Parameters:
data – The tensor providing input data.
axis – An axis along which Softmax should be calculated
- Returns:
The new node with softmax operation applied on each element.