Skip to main content

//clika-runtime/io.clika.runtime/F

F

[jvm]
object F

The functional namespace: stateless spellings of the runtime ops, keyword-defaulted per the binding signature conventions.

val a = F.softmax(logits, dim = -1)
val h = F.linear(x, weight, bias)
val r = F.relu(h)

Every call is NON-CONSUMING (the retain-before-consuming-op mechanism) and returns a fresh caller-owned Tensor.

Functions

NameSummary
amax[jvm]
@JvmOverloads
fun amax(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false): Tensor
amax(input, dims=[], keepdim=false); the global max reduce.
gelu[jvm]
@JvmOverloads
fun gelu(input: Tensor, approximate: Int = ClikaRtGen.GELU_MODE_NONE): Tensor
gelu(input, approximate=none); approximate is a ClikaRtGen.GELU_MODE_* value (tanh / quick / fast forms).
linear[jvm]
@JvmOverloads
fun linear(input: Tensor, weight: Tensor, bias: Tensor? = null, activation: Int? = null): Tensor
linear(input, weight, bias=null): input @ weight^T + bias. The fused activation is keyword-only-shaped: pass a ClikaRtGen.ACTIVATION_* value to fuse one; the plain call reads like the plain op.
logSoftmax[jvm]
@JvmOverloads
fun logSoftmax(input: Tensor, dim: Long = -1, dtype: Int = ClikaRtGen.DATA_TYPE_UNDEFINED): Tensor
logSoftmax(input, dim=-1, dtype=undefined).
matmul[jvm]
fun matmul(input: Tensor, other: Tensor): Tensor
matmul(input, other): the function form of the infix operator.
mean[jvm]
@JvmOverloads
fun mean(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false, dtype: Int = ClikaRtGen.DATA_TYPE_UNDEFINED): Tensor
mean(input, dims=[], keepdim=false); empty dims reduce all.
relu[jvm]
fun relu(input: Tensor): Tensor
relu(input), elementwise.
sigmoid[jvm]
fun sigmoid(input: Tensor): Tensor
sigmoid(input), elementwise.
silu[jvm]
fun silu(input: Tensor): Tensor
silu(input), elementwise.
softmax[jvm]
@JvmOverloads
fun softmax(input: Tensor, dim: Long = -1, dtype: Int = ClikaRtGen.DATA_TYPE_UNDEFINED): Tensor
softmax(input, dim=-1, dtype=undefined); an undefined dtype keeps the input's dtype.
sum[jvm]
@JvmOverloads
fun sum(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false, dtype: Int = ClikaRtGen.DATA_TYPE_UNDEFINED): Tensor
sum(input, dims=[], keepdim=false); empty dims reduce all.
tanh[jvm]
fun tanh(input: Tensor): Tensor
tanh(input), elementwise.