Mojo function
mean
mean(src: LayoutTensor[dtype, layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment]) -> Scalar[dtype]
Computes the mean value of the elements in a buffer.
Args:
- src (
LayoutTensor
): The buffer of elements for which the mean is computed.
Returns:
Scalar
: The mean value of the elements in the given buffer.
mean[reduce_axis: Int](src: LayoutTensor[dtype, layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment], dst: LayoutTensor[dtype, layout, origin, address_space=address_space, element_layout=element_layout, layout_int_type=layout_int_type, linear_idx_type=linear_idx_type, masked=masked, alignment=alignment])
Computes the mean across reduce_axis of an NDBuffer.
Parameters:
- reduce_axis (
Int
): The axis to reduce across.
Args:
- src (
LayoutTensor
): The input buffer. - dst (
LayoutTensor
): The output buffer.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!