@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
10 lines • 334 B
JavaScript
import { UnaryNode } from './unaryNode';
export class RoundNode extends UnaryNode {
constructor(attributes, inputs, outputs, constants, onnxVersion, mode) {
super(attributes, inputs, outputs, constants, onnxVersion, 'Round', mode);
}
compute(x) {
return x.round();
}
}
//# sourceMappingURL=round.js.map