0afa83b99fccef7be77ec5bb3fb1321f5a4b4dd6,thinc/backends/ops.py,Ops,backprop_take,#Ops#Any#Any#Any#,269

Before Change

        return output

    def backprop_take(self, dX: Array, which: Array, nP: int) -> Array:
        dX__bop = self.allocate((dX.shape[0], dX.shape[1], nP))
        for i in range(nP):
            dX__bop[:, :, i] += dX * (which == i)
        return dX__bop

After Change

        return output

    def backprop_take(self, dX: Array, which: Array, nP: int) -> Array:
        dX__bop = self.alloc_f3d(dX.shape[0], dX.shape[1], nP)
        for i in range(nP):
            dX__bop[:, :, i] += dX * (which == i)
        return dX__bop

Instances