TypeTensor - v0.1.0
Preparing search index...
@typetensor/core
LogSoftmaxOp
Type Alias LogSoftmaxOp<Input, Axis, OutputDType>
LogSoftmaxOp
:
ValidateDim
<
Axis
,
Input
[
"__shape"
]
>
extends
DimensionError
<
string
>
?
never
:
StorageTransformation
<
"log_softmax"
,
TensorStorage
<
OutputDType
,
Input
[
"__shape"
]
,
ComputeStrides
<
Input
[
"__shape"
]
>
|
Input
[
"__strides"
]
,
SoftmaxOpLayout
<
Input
[
"__layout"
]
>
,
>
,
readonly
[
Input
]
,
>
&
{
__logSoftmaxAxis
:
ValidateDim
<
Axis
,
Input
[
"__shape"
]
>
}
Log-softmax operation interface Computes log(softmax(x)) for numerical stability
Type Parameters
Input
extends
AnyTensorStorage
Axis
extends
number
OutputDType
extends
AnyDType
=
ToFloat
<
Input
[
"__dtype"
]
>
Settings
Member Visibility
Protected
Inherited
External
Theme
OS
Light
Dark
TypeTensor - v0.1.0
Loading...
Log-softmax operation interface Computes log(softmax(x)) for numerical stability