@hoff97/tensor-js/dist/lib/autograd/ops/unary/sqrtBack.js

PyTorch like deep learning inferrence library

export class SqrtBack {
    constructor(input, sqrt) {
        this.input = input;
        this.sqrt = sqrt;
    }
    backward(grad) {
        const gradSqrt = grad.divide(this.sqrt, 0.5);
        const needed = this.input.backward(gradSqrt);
        if (!needed) {
            gradSqrt.delete();
        }
    }
    delete() {
        if (!this.input.isLeaf()) {
            this.input.delete();
        }
    }
}
//# sourceMappingURL=sqrtBack.js.map