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

Data analysis model package without any dependencies

/**
 * Split layer
 */
export default class SplitLayer extends Layer {
    /**
     * @param {object} config config
     * @param {number} [config.axis] axis
     * @param {number | number[]} config.size size
     */
    constructor({ axis, size, ...rest }: {
        axis?: number;
        size: number | number[];
    });
    _axis: number;
    _size: number | number[];
    calc(x: any): any[];
    _o: any[];
    grad(...bo: any[]): any;
    toObject(): {
        type: string;
        axis: number;
        size: number | number[];
    };
}
import Layer from './base.js';