Function arm_mean_s16

Function Documentation

arm_cmsis_nn_status arm_mean_s16(const int16_t *input_data, const cmsis_nn_dims *input_dims, const int32_t input_offset, const cmsis_nn_dims *axis_dims, int16_t *output_data, const cmsis_nn_dims *output_dims, const int32_t out_offset, const int32_t out_mult, const int32_t out_shift)

Computes the mean of the input tensor along the specified axis. The output multipler and shift must have the output count folded into them.

Parameters:

  • input_data – [in] Pointer to input tensor

  • input_dims – [in] Input tensor dimensions

  • input_offset – [in] Input offset

  • axis_dims – [in] Axis dimensions to compute mean over

  • output_data – [out] Pointer to output tensor

  • output_dims – [in] Output tensor dimensions

  • out_offset – [in] Output offset

  • out_mult – [in] Output quantization multiplier

  • out_shift – [in] Output quantization shift

Returns:

The function returns ARM_CMSIS_NN_SUCCESS