Julia使用argmax返回矩阵行内的索引，而不是笛卡尔索引

假设我有一个矩阵

X = [1 2 3 4; 1 4 3 2];

我想找到矩阵中每一行的argmax，相对于该行，而不是整个X中条目的索引。这意味着我希望argmax（X，dims=（2））的输出是一个向量

[4, 2];

但当前的输出是笛卡尔指数数组

[CartesianIndex(1, 4), CartesianIndex(2, 2)];

---

有没有办法在argmax函数中指定这一点，或者有效地将输出转换为我想要的状态？

您可以使用

eachrow

遍历矩阵的行：

julia> argmax.(eachrow(X))
2-element Vector{Int64}:
 4
 2