nn_log_softmax {torch} | R Documentation |
LogSoftmax module
Description
Applies the function to an n-dimensional
input Tensor. The LogSoftmax formulation can be simplified as:
Usage
nn_log_softmax(dim)
Arguments
dim |
(int): A dimension along which LogSoftmax will be computed. |
Details
Value
a Tensor of the same dimension and shape as the input with values in the range [-inf, 0)
Shape
Input:
where
*
means, any number of additional dimensionsOutput:
, same shape as the input
Examples
if (torch_is_installed()) {
m <- nn_log_softmax(1)
input <- torch_randn(2, 3)
output <- m(input)
}
[Package torch version 0.13.0 Index]