UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

13 lines 482 B
import { ReduceNode } from './reduceNode'; export class ReduceL2Node extends ReduceNode { constructor(attributes, inputs, outputs, constants, onnxVersion, mode) { super(attributes, inputs, outputs, constants, onnxVersion, 'ReduceL2', mode); } calc(input) { const sumSquare = input.sumSquare(this.axes, this.keepDims); const result = sumSquare.sqrt(); sumSquare.delete(); return result; } } //# sourceMappingURL=reduceL2.js.map