ngraph.opset1.ops.softmax

ngraph.opset1.ops.softmax(data: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: int, name: Optional[str] = None)_pyngraph.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.