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

Data analysis model package without any dependencies

/**
 * Softargmax layer
 */
export default class SoftargmaxLayer extends Layer {
    /**
     * @param {object} config config
     * @param {number} [config.beta] beta
     */
    constructor({ beta, ...rest }: {
        beta?: number;
    });
    _beta: number;
    calc(x: any): Matrix<number>;
    _i: any;
    _o: Matrix<number>;
    grad(bo: any): Matrix<T>;
    _bo: Matrix<T>;
    _bi: Matrix<T>;
    toObject(): {
        type: string;
        beta: number;
    };
}
import Layer from './base.js';
import Matrix from '../../../util/matrix.js';