f4559a3d47510f776b3e38d5aceb07ee4b4b0ea9,gpflow/transforms.py,DiagMatrix,backward,#DiagMatrix#Any#,277

Before Change

    def backward(self, y):
        // Return diagonals of matrices
        x = y.reshape(-1, self.dim, self.dim).diagonal(0, 1, 2).flatten()
        return self._positive_transform.backward(x)

    def forward_tensor(self, x):
        y = self._positive_transform.forward_tensor(x)

After Change

    def backward(self, y):
        // Return diagonals of matrices
        if not (y.shape[1] == y.shape[2] == self.dim) and (len(y.shape) == 3):
            raise ValueError("shape of input does not match this transform")
        return y.diagonal(offset=0, axis1=1, axis2=2).flatten()

    def forward_tensor(self, x):
        // create diagonal; matrices

Instances