Method _bwd

numpy_ml/neural_nets/layers/layers.py:739–742

Actual computation of gradient of the loss wrt. each input

(self, dLdY, X, _sum)

        return grads[0] if len(X) == 1 else grads

    def _bwd(self, dLdY, X, _sum):
        """Actual computation of gradient of the loss wrt. each input"""
        grads = [dLdY * self.act_fn.grad(_sum) for _ in X]
        return grads


class Multiply(LayerBase):