8f27fae607562177d6b840a0c0db2c67a95a88d8,gpytorch/likelihoods/softmax_likelihood.py,SoftmaxLikelihood,forward,#SoftmaxLikelihood#Any#,24

Before Change

        mixed_fs = self.mixing_weights.matmul(function_samples)  // num_classes x num_data
        mixed_fs = mixed_fs.transpose(-1, -2)  // num_data x num_classes
        softmax = torch.nn.functional.softmax(mixed_fs, -1)
        return base_distributions.Categorical(probs=softmax)

After Change

        mixed_fs = self.mixing_weights @ function_samples  // num_classes x num_data
        mixed_fs = mixed_fs.transpose(-1, -2)  // num_data x num_classes
        res = base_distributions.Categorical(logits=mixed_fs)
        return res

Instances