nnf_log_softmax: Log_softmax
In torch: Tensors and Neural Networks with 'GPU' Acceleration
View source: R/nnf-activation.R
nnf_log_softmax R Documentation
Log_softmax
Description
Applies a softmax followed by a logarithm.
Usage
nnf_log_softmax(input, dim = NULL, dtype = NULL)
Arguments
input
(Tensor) input
dim
(int) A dimension along which log_softmax will be computed.
dtype
(torch.dtype, optional) the desired data type of returned tensor.
If specified, the input tensor is casted to dtype before the operation
is performed. This is useful for preventing data type overflows.
Default: NULL.
Details
While mathematically equivalent to log(softmax(x)), doing these two
operations separately is slower, and numerically unstable. This function
uses an alternative formulation to compute the output and gradient correctly.
torch documentation built on May 29, 2024, 9:54 a.m.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
View source: R/nnf-activation.R
| nnf_log_softmax | R Documentation |
Log_softmax
Description
Applies a softmax followed by a logarithm.
Usage
nnf_log_softmax(input, dim = NULL, dtype = NULL)
Arguments
input |
(Tensor) input |
dim |
(int) A dimension along which log_softmax will be computed. |
dtype |
( |
Details
While mathematically equivalent to log(softmax(x)), doing these two operations separately is slower, and numerically unstable. This function uses an alternative formulation to compute the output and gradient correctly.
R Package Documentation
Browse R Packages
We want your feedback!
Note that we can't provide technical support on individual packages. You should contact the package authors for that.
Embedding an R snippet on your website
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.