openvino.runtime.opset11.softmax#

openvino.runtime.opset11.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. Can be positive or negative.

  • name – Optional name for the node.

Returns:

The new node with softmax operation applied on each element.