@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
13 lines • 482 B
JavaScript
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