@ai-on-browser/data-analysis-models/lib/model/nns/layer/argmin.d.ts

Data analysis model package without any dependencies

/**
 * Argmin layer
 */
export default class ArgminLayer extends Layer {
    /**
     * @param {object} config object
     * @param {number} [config.axis] Axis
     * @param {boolean} [config.keepdims] Keep dimensions
     */
    constructor({ axis, keepdims, ...rest }: {
        axis?: number;
        keepdims?: boolean;
    });
    _axis: number;
    _keepdims: boolean;
    calc(x: any): any;
    _i: any;
    _o: any;
    grad(bo: any): any;
    _bo: any;
    _bi: any;
    toObject(): {
        type: string;
        axis: number;
        keepdims: boolean;
    };
}
import Layer from './base.js';