Syslab中的max函数无法返回最大值对应的下标，可以使用argmax函数返回最大值的下标，示例如下：

argmax(A)返回集合A最大值的索引。可以有多个最大元素，在这种情况下argmax将返回一个最大索引，但不一定是第一个。

返回向量最大值的索引：

返回矩阵最大值的索引：

argmax(A;dims) 返回集合A沿维度dims的最大值索引。

返回A沿列的最大值索引：

返回A沿行的最大值索引：