argmax¶

paddle. argmax ( x, axis=None, keepdim=False, dtype='int64', name=None ) [源代码] ¶

该OP沿 axis 计算输入 x 的最大元素的索引。

参数¶

x （Tensor） - 输入的多维 Tensor ，支持的数据类型：float32、float64、int16、int32、int64、uint8。

axis （int，可选） - 指定对输入Tensor进行运算的轴， axis 的有效范围是[-R, R），R是输入 x 的维度个数， axis 为负数时，进行计算的 axis 与 axis + R 一致。默认值为None, 将会对输入的 x 进行平铺展开，返回最大值的索引。

keepdim （bool，可选）- 是否保留进行最大值索引操作的轴，默认值为False。

dtype （np.dtype|str，可选）- 输出Tensor的数据类型，可选值为int32，int64，默认值为int64，将返回int64类型的结果。

name （str，可选） – 具体用法请参见 Name ，一般无需设置，默认值为None。

返回¶

Tensor, 如果设置 dtype 为 int32 时，返回的tensor的数据类型为 int32 ，其它情况将返回的tensor的数据类型为 int64 。

示例代码¶

import numpy as np
import paddle

data = [[5,8,9,5],
        [0,0,1,7],
        [6,9,2,4]]
x =  paddle.to_tensor(data)
out1 = paddle.argmax(x)
print(out1) # 2
out2 = paddle.argmax(x, axis=1)
print(out2)
# [2 3 1]
out3 = paddle.argmax(x, axis=-1)
print(out3)
# [2 3 1]

PaddlePaddle

paddle / argmax

argmax¶

参数¶

返回¶

示例代码¶

PaddlePaddle

paddle

paddle.amp

paddle.callbacks

paddle.compat

paddle.distributed

paddle.distribution

paddle.fluid

paddle.io

paddle.jit

paddle.metric

paddle.nn

paddle.onnx

paddle.optimizer

paddle.regularizer

paddle.static

paddle.sysconfig

paddle.text

paddle.utils

paddle.vision

paddle / argmax

argmax¶

参数¶

返回¶

示例代码¶