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
- Return:
The new node with LogSoftmax operation applied on each element.