@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
9 lines (8 loc) • 452 B
TypeScript
import { Mode } from '../../../model/module';
import Tensor, { DType } from '../../../types';
import { Attributes, Constants } from '../../types';
import { ReduceNode } from './reduceNode';
export declare class ReduceSumSquareNode extends ReduceNode {
constructor(attributes: Attributes, inputs: string[], outputs: string[], constants: Constants, onnxVersion: number, mode: Mode);
calc<DTpe extends DType>(input: Tensor<DTpe>): Tensor<DTpe>;
}