Function GetOutResult

executor/operator/ref/kernel/softmax/ref_softmax_kernel.h:62–94

}

static void GetOutResult(float* input, float* output, float* array, float* sum_array, int in_size, int on_size)
{
    float* input_ptr = ( float* )input;
    float* output_ptr = ( float* )output;
    float* array_ptr = ( float* )array;
    float* sum_array_ptr = ( float* )sum_array;

    memset(sum_array, 0x0, in_size * sizeof(float));

    /* get the exp and the summary */

    for(int j = 0; j < on_size; j++)
    {
       
        for(int l = 0; l < in_size; l++)
        {
            int index = j * in_size + l;
            output_ptr[index] = exp(input_ptr[index] - array_ptr[l]);
            sum_array_ptr[l] += output_ptr[index];
           
        }
    }
     
    /* the final result */
    for(int j = 0; j < on_size; j++)
    {
        for(int l = 0; l < in_size; l++)
        {
            int index = j * in_size + l;
            output_ptr[index] /= sum_array_ptr[l];
        }
    }
}

typedef int (*ref_softmax_kernel_t)(void* input, void* output, void* max_array, void* sum_array, op_data* op_param);