Function autograd::ops::log_softmax
[−]
[src]
pub fn log_softmax(x: &Tensor, axis: isize) -> Tensor
Log softmax function.
Computes softmax(x)
along specified axis and
takes logarithm of it.
axis
can be negative.