@hoff97/tensor-js
Version:
PyTorch like deep learning inferrence library
18 lines • 417 B
JavaScript
export class LogBack {
constructor(input) {
this.input = input;
}
backward(grad) {
const gradLog = grad.divide(this.input.value);
const needed = this.input.backward(gradLog);
if (!needed) {
gradLog.delete();
}
}
delete() {
if (!this.input.isLeaf()) {
this.input.delete();
}
}
}
//# sourceMappingURL=logBack.js.map