UNPKG

@hoff97/tensor-js

Version:

PyTorch like deep learning inferrence library

18 lines 417 B
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