//clika-runtime/io.clika.runtime/Ops/softmax

softmax

[common]
fun softmax(input: Tensor, dim: Long = -1L, dtype: DType = DType.UNDEFINED): Tensor

softmax(input: Tensor, dim: Long = -1L, dtype: DType = DType.UNDEFINED): the softmax operator. Softmax along dim, computed stably (max-subtracted).