（beta）torch_npu.npu_max

函数原型

功能说明

使用dim对最大结果进行计算。类似于torch.max，优化NPU设备实现。

参数说明

• self (Tensor) - 输入张量。
• dim (Int) - 待降低维度。
• keepdim (Bool，默认值为False) - 输出张量是否保留dim。

输出说明

• values (Tensor) - 输入张量中的最大值。
• indices (Tensor) - 输入张量中最大值的index。

支持的型号

• [object Object]Atlas 训练系列产品[object Object]
• [object Object]Atlas A2 训练系列产品[object Object]
• [object Object]Atlas A3 训练系列产品[object Object]
• [object Object]Atlas 推理系列产品[object Object]

调用示例