argmax

    argmax

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

    返回类型:Variable

    代码示例