nn_log_softmax function

LogSoftmax module

Applies the $\log(\mbox{Softmax}(x))$ function to an n-dimensional input Tensor. The LogSoftmax formulation can be simplified as:

nn_log_softmax(dim)

Arguments

• dim: (int): A dimension along which LogSoftmax will be computed.

Returns

a Tensor of the same dimension and shape as the input with values in the range [-inf, 0)

Details

Shape

• Input: $(*)$ where * means, any number of additional dimensions
• Output: $(*)$, same shape as the input

Examples

if (torch_is_installed()) {
m <- nn_log_softmax(1)
input <- torch_randn(2, 3)
output <- m(input)
}