fusedRmsNorm

abstract fun <T : DType, V> fusedRmsNorm(input: Tensor<T, V>, weight: Tensor<T, V>, eps: Float): Tensor<T, V>?(source)

Compute fused RMS normalization: outputi = (inputi / rms(input)) * weighti

Return

result tensor, or null if this implementation cannot handle the input