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

PyTorch like deep learning inferrence library

import { Mode } from '../../model/module';
import Tensor from '../../types';
import { OnnxNode } from '../node';
import { Attributes, Constants } from '../types';
export declare class GatherNode extends OnnxNode {
    private axis;
    constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode);
    forward(inputs: Tensor<any>[]): Promise<Tensor<any>[]>;
    getType(): string;
    delete(): void;
}