argMax 以数组为轴做什么？

Question

在tensorflowjs的教程中，对于recognizing handwritten numbers，代码使用：

const labels = testData.labels.argMax([-1]);

然而，Tensor<Rank.R2>.argMax 似乎以一个轴作为参数，它是一个数字（也有道理）。打字稿相应地抱怨：

这里发生了什么？这只是教程中的一个错误，argMax只是将一个元素数组转换为数字，得到它所需要的吗？这里有什么约定或更深层次的含义吗？

一般来说，argmax 操作应该是可交换的，它可以解释例如.argMax([-1, -2]) 和 .argMax(-1).argMax(-1) 一样，然而，情况似乎并非如此。

Answer 1

axis 的 ts 类型表明它是一个数字，因此 tslint 的抱怨。 查看源代码，轴稍后被推入一个数组，其中包含

[].concat(axis)

无论是数字还是单个数字的数组，都会return一样。但是目前还不支持多轴数组。