@tensorflow/tfjs-core/dist/public/chained_ops/arg_max.d.ts

Hardware-accelerated JavaScript library for machine intelligence

import { Rank } from '../../types';
declare module '../../tensor' {
    interface Tensor<R extends Rank = Rank> {
        argMax<T extends Tensor>(axis?: number): T;
    }
}