使用dim对最大结果进行计算。类似于torch.max,优化NPU设备实现。
[object Object]
- self (
[object Object]):必选参数,输入张量。 - dim (
[object Object]):必选参数,待降低维度。 - keepdim (
[object Object]):可选参数,输出张量是否保留dim。默认值为False。
- values (
[object Object]):输入张量中的最大值。 - indices (
[object Object]):输入张量中最大值的index。
[object Object]