59562fbb1d97a00c04719b9a8e2b9cf69df54571,qucumber/rbm/purification_rbm.py,PurificationRBM,prob_a_given_v,#PurificationRBM#Any#Any#,215

Before Change

                  given the visible state
        :rtype torch.Tensor:
        
        return (
            torch.matmul(v, self.weights_U.data.t(), out=out)
            .add_(self.aux_bias.data)
            .sigmoid_()
        )

    @auto_unsqueeze_arg(1, 2)
    def prob_v_given_ha(self, h, a, out=None):
        rGiven a hidden and auxiliary unit configuration, compute

After Change

                  given the visible state
        :rtype torch.Tensor:
        
        return (
            torch.matmul(v, self.weights_U.data.t(), out=out)
            .add_(self.aux_bias.data)
            .sigmoid_()
            .clamp_(min=0, max=1)
        )

    @auto_unsqueeze_arg(1, 2)
    def prob_v_given_ha(self, h, a, out=None):
        rGiven a hidden and auxiliary unit configuration, compute

Instances