# 比较与tf.arg_max的功能差异

[![查看源文件](https://mindspore-website.obs.cn-north-4.myhuaweicloud.com/website-images/r1.8/resource/_static/logo_source.png)](https://gitee.com/mindspore/docs/blob/r1.8/docs/mindspore/source_zh_cn/note/api_mapping/tensorflow_diff/TensorArgmax.md)

## tf.arg_max

```python
tf.arg_max(input, dimension, output_type=tf.dtypes.int64, name=None)
```

更多内容详见[tf.arg_max](https://www.tensorflow.org/versions/r1.15/api_docs/python/tf/arg_max)。

## mindspore.Tensor.argmax

```python
mindspore.Tensor.argmax(axis=None)
```

更多内容详见[mindspore.Tensor.argmax](https://mindspore.cn/docs/zh-CN/r1.8/api_python/mindspore/mindspore.Tensor.html#mindspore.Tensor.argmax)。

## 使用方式

功能相同，MindSpore和TensorFlow的两接口分别通过参数`axis`和`dimension`入参决定基于哪个维度返回最大值的索引。

不同点在于，默认状态下，MindSpore的`axis=None`，返回最大值的全局索引；TensorFlow的`dimension`不传入数值时，默认返回`dimension=0`的最大值索引。

## 代码示例

```python
import mindspore as ms

a = ms.Tensor([[1, 10, 166.32, 62.3], [1, -5, 2, 200]], ms.float32)
print(a.argmax())
print(a.argmax(axis=0))
print(a.argmax(axis=1))
# output:
# 7
# [0 0 0 1]
# [2 3]

import tensorflow as tf
tf.enable_eager_execution()

b = tf.constant([[1, 10, 166.32, 62.3], [1, -5, 2, 200]])
print(tf.argmax(b).numpy())
print(tf.argmax(b, dimension=0).numpy())
print(tf.argmax(b, dimension=1).numpy())
# output:
# [0 0 0 1]
# [0 0 0 1]
# [2 3]
```