For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).

Mojo function

row_mean_of_squares

def row_mean_of_squares[in_dtype: DType, out_dtype: DType, rank: Int, InputFn: def[width: Int, rank: Int](IndexList[rank]) -> SIMD[in_dtype, width] & RegisterPassable & ImplicitlyCopyable, OutputFn: def[width: SIMDLength, rank: Int](IndexList[rank], SIMD[out_dtype, width]) -> None & RegisterPassable & ImplicitlyCopyable, /, target: StringSpan[ImmStaticOrigin] = StringSpan("cpu"), reduce_dim: Int = (rank - Int(1))](input_fn: InputFn, output_fn: OutputFn, shape: IndexList[rank], ctx: DeviceContext) where identical(InputFn.in_dtype, in_dtype) where identical(OutputFn.out_dtype, out_dtype)