296b28c1a5b1fb18732d1ed0f466b3761e50ebef,pyro/contrib/gp/kernels/coregionalize.py,Coregionalize,__init__,#Coregionalize#Any#Any#Any#Any#Any#,44

Before Change

        if diagonal.shape != (input_dim,):
            raise ValueError("Expected diagonal.shape == ({},), actual {}"
                             .format(input_dim, diagonal.shape))
        self.diagonal = Parameter(diagonal)
        self.set_constraint("diagonal", constraints.positive)

    def forward(self, X, Z=None, diag=False):
        X = self._slice_input(X)
        Xc = X.matmul(self.components)

After Change

        if diagonal.shape != (input_dim,):
            raise ValueError("Expected diagonal.shape == ({},), actual {}"
                             .format(input_dim, diagonal.shape))
        self.diagonal = PyroParam(diagonal, constraints.positive)

    def forward(self, X, Z=None, diag=False):
        X = self._slice_input(X)
        Xc = X.matmul(self.components)

Instances