openvino.runtime.opset14.softmax

openvino.runtime.opset14.softmax(data: Union[openvino._pyopenvino.Node, int, float, numpy.ndarray], axis: int, name: Optional[str] = None) → openvino._pyopenvino.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. Can be positive or negative.

• name – Optional name for the node.

Returns

The new node with softmax operation applied on each element.