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

PyTorch like deep learning inferrence library

export class SigmoidBack {
    constructor(input, sigmoid) {
        this.input = input;
        this.sigmoid = sigmoid;
    }
    backward(grad) {
        const oneMinus = this.sigmoid.addMultiplyScalar(-1, 1);
        const mult = this.sigmoid.multiply(oneMinus);
        oneMinus.delete();
        const gradIn = mult.multiply(grad);
        mult.delete();
        const needed = this.input.backward(gradIn);
        if (!needed) {
            gradIn.delete();
        }
    }
    delete() {
        if (!this.input.isLeaf()) {
            this.input.delete();
        }
    }
}
//# sourceMappingURL=sigmoidBack.js.map