Function argmax

argmax(tensor: Tensor, axis: number, keep_dims?: boolean): Tensor
Parameters
- tensor: Tensor
- axis: number
- keep_dims: boolean = false
Returns Tensor
- Defined in shumai/tensor/tensor_ops_gen.ts:3168

Generated using TypeDoc