@hoff97/tensor-js/dist/lib/onnx/nodes/reduce/reduceNode.d.ts

PyTorch like deep learning inferrence library

import { Tensor } from '../../../library';
import { Mode } from '../../../model/module';
import { DType } from '../../../types';
import { OnnxNode } from '../../node';
import { Attributes, Constants } from '../../types';
export declare abstract class ReduceNode extends OnnxNode {
    protected axes?: number[];
    protected keepDims?: boolean;
    protected name: string;
    constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, name: string, mode: Mode);
    abstract calc<DTpe extends DType>(input: Tensor<DTpe>): Tensor<DTpe>;
    protected getAxes<DTpe extends DType>(input: Tensor<DTpe>): any[];
    forward<DTpe extends DType>(inputs: Tensor<DTpe>[]): Promise<Tensor<DTpe>[]>;
    getType(): string;
    delete(): void;
}