g++ split-tensor.cpp -o split-tensor
g++ find_argmax_of_tensor_FP32.cpp -o find_argmax_of_tensor_FP32