SKaiNET
Toggle table of contents
jvm
Target filter
jvm
Switch theme
Search in API
Skip to content
SKaiNET
skainet-lang-core
/
sk.ainet.java
/
TensorJavaOps
/
softmax
softmax
jvm
@
JvmStatic
@
JvmOverloads
fun
softmax
(
a
:
Tensor
<
*
,
*
>
,
dim
:
Int
=
-1
)
:
Tensor
<
*
,
*
>
(
source
)
Softmax along the given dimension.