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

Data analysis model package without any dependencies

/**
 * Readout layer
 */
export default class ReadoutLayer extends Layer {
    /**
     * @param {object} config config
     * @param {'sum' | 'mean'} [config.method] Aggregate method
     */
    constructor({ method, ...rest }: {
        method?: "sum" | "mean";
    });
    _method: "sum" | "mean";
    calc(x: any): Tensor<number> | Matrix<number>;
    _i: any;
    _o: Tensor<number> | Matrix<number>;
    grad(bo: any): any;
    _bi: any;
    toObject(): {
        type: string;
        method: "sum" | "mean";
    };
}
import Layer from './base.js';
import Tensor from '../../../util/tensor.js';
import Matrix from '../../../util/matrix.js';