UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

13 lines (12 loc) 638 B
import { Mode } from '../../../model/module'; import Tensor, { DType } from '../../../types'; import { OnnxNode } from '../../node'; import { Attributes, Constants } from '../../types'; export declare abstract class BinaryNode extends OnnxNode { protected name: string; constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, name: string, mode: Mode); abstract compute<DTpe extends DType>(a: Tensor<DTpe>, b: Tensor<DTpe>): Tensor<DTpe>; forward<DTpe extends DType>(inputs: Tensor<DTpe>[]): Promise<Tensor<DTpe>[]>; getType(): string; delete(): void; }