argmax

    该OP沿 axis 计算输入 x 的最大元素的索引。

    , 如果设置 为 int32 时，返回的tensor的数据类型为 int32 ，其它情况将返回的tensor的数据类型为 int64 。