@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
18 lines • 404 B
JavaScript
export class NegateBack {
constructor(input) {
this.input = input;
}
backward(grad) {
const gradIn = grad.negate();
const needed = this.input.backward(gradIn);
if (!needed) {
gradIn.delete();
}
}
delete() {
if (!this.input.isLeaf()) {
this.input.delete();
}
}
}
//# sourceMappingURL=negateBack.js.map