import * as tf from '@tensorflow/tfjs-core';
export declare function argmax2d(inputs: tf.Tensor3D): tf.Tensor2D;