Skip to main content

//clika-runtime/io.clika.runtime/Ops/mean

mean

[common]
fun mean(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false, dtype: DType = DType.UNDEFINED): Tensor

mean(input: Tensor, dims: LongArray = longArrayOf(), keepdim: Boolean = false, dtype: DType = DType.UNDEFINED): the mean operator. Arithmetic mean of input over dims. Empty dims averages every element. dtype selects the output (and accumulation) dtype; the default keeps input's.