mindspore.mint.amax ======================== .. image:: https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/master/resource/_static/logo_source.svg :target: https://gitee.com/mindspore/mindspore/blob/master/docs/api/api_python/mint/mindspore.mint.func_amax.rst :alt: 查看源文件 .. py:function:: mindspore.mint.amax(input, dim=(), keepdim=False) 返回tensor在指定维度上的最大值。 .. warning:: 这是一个实验性API,后续可能修改或删除。 参数: - **input** (Tensor) - 输入tensor。 - **dim** (Union[int, tuple(int), list(int)], 可选) - 指定计算维度。 当 `dim` 为 `()` 时,计算 `input` 中的所有元素。默认 ``()``。 - **keepdim** (bool, 可选) - 输出tensor是否保留维度。默认 ``False``。 返回: Tensor