开发者
资源
[object Object][object Object][object Object]undefined
[object Object]

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

[object Object]
[object Object]
[object Object]
  • self ([object Object]):必选参数,输入张量。
  • dim ([object Object]):必选参数,待降低维度。
  • keepdim ([object Object]):可选参数,输出张量是否保留dim。默认值为False。
[object Object]
  • values ([object Object]):输入张量中的最大值。
  • indices ([object Object]):输入张量中最大值的index。
[object Object]
[object Object]