在使用mask.max(dim=2)时会用到argmax 求最大值索引，但是在转onnx为om时就报argmax算子不支持，请问怎么解决？