f065b29cc8667c3012761b9509886411d8095e57,starry/_core/ops/integration.py,sTReflectedGradientOp,perform,#sTReflectedGradientOp#Any#Any#Any#,142

Before Change

    def perform(self, node, inputs, outputs):
        b, theta, bo, ro, bsT = inputs
        bb, btheta, bbo, bro = self.base_op.func(b, theta, bo, ro, bsT)
        outputs[0][0] = np.reshape(bb, np.shape(inputs[0]))
        outputs[1][0] = np.reshape(btheta, np.shape(inputs[1]))
        outputs[2][0] = np.reshape(bbo, np.shape(inputs[2]))
        outputs[3][0] = np.reshape(bro, np.shape(inputs[3]))

After Change

        bb = (bsT * ddb).sum(-1)
        btheta = (bsT * ddtheta).sum(-1)
        bbo = (bsT * ddbo).sum(-1)
        bro = (bsT * ddro).sum()
        outputs[0][0] = np.reshape(bb, np.shape(b))
        outputs[1][0] = np.reshape(btheta, np.shape(theta))
        outputs[2][0] = np.reshape(bbo, np.shape(bo))
        outputs[3][0] = np.reshape(bro, np.shape(ro))

Instances