arg_max.h

#include "common.h"
 
#define FLOAT_MIN -1000000.0
 
void arg_max_dim2(Matrix3D<float> &input, Matrix3D<int> &output);