10bdd46d76d35cfba3a484b007ea3e14af801813,qucumber/rbm/purification_rbm.py,PurificationRBM,prob_a_given_v,#PurificationRBM#,195

Before Change

                  given the visible state
        :rtype torch.Tensor:
        
        if v.dim() < 2:  // create extra axis, if needed
            v = v.unsqueeze(0)
            unsqueezed = True
        else:
            unsqueezed = False

        p = torch.addmm(
            self.aux_bias.data, v, self.weights_U.data.t(), out=out
        ).sigmoid_()

        if unsqueezed:
            return p.squeeze_(0)  // remove superfluous axis, if it exists
        else:
            return p

    def prob_v_given_ha(self, h, a, out=None):
        rGiven a hidden and auxiliary unit configuration, compute
        the probability vector of the hidden units being on

After Change

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

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

Instances