Edit this page

ngraph.log_softmax

ngraph.log_softmax(data: Union[_pyngraph.Node, int, float, numpy.ndarray], axis: int, name: Optional[str] = None) → _pyngraph.Node

Apply LogSoftmax operation on each element of input tensor.

Parameters

• data – The tensor providing input data.

• axis – An axis along which LogSoftmax should be calculated

Returns

The new node with LogSoftmax operation applied on each element.

Prev Next