b10173655589b038ba1e69e937eddf03819dc94d,spacy/_ml.py,PrecomputableAffine,begin_update,#PrecomputableAffine#Any#Any#,135

Before Change

        return self.nF * self.nO

    def begin_update(self, X, drop=0.):
        nN = X.shape[0]
        // X: (b, i)
        // Xf: (b, f, i)
        // Yf: (b, f, o)
        // dY: (b, o)

After Change

        tensordot = self.ops.xp.tensordot
        ascontiguous = self.ops.xp.ascontiguousarray

        Yf = tensordot(X, self.W, axes=[[1], [2]])

        def backward(dY_ids, sgd=None):
            dY, ids = dY_ids
            Xf = X[ids]

Instances