paddle源码paddle.argmax(x, axis=None, keepdim=False, dtype='int64', name=None) 其中keepdim(bool,可选)- 是否在输出Tensor中保留减小的维度。如果 keepdim 为True,则输出Tensor和 x 具有相同的维度(减少的维度除外,减少的维度的大小为1),默认值为False。 tensorlayerx源码paddle.argmax(x, axis=None, dtype='int64')