openvino.runtime.opset8.softmax¶
-
openvino.runtime.opset8.
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.