10bdd46d76d35cfba3a484b007ea3e14af801813,qucumber/rbm/binary_rbm.py,BinaryRBM,prob_v_given_h,#BinaryRBM#,133

Before Change

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

        p = torch.addmm(
            self.visible_bias.data, h, self.weights.data, out=out
        ).sigmoid_()

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

    def prob_h_given_v(self, v, out=None):
        Given a visible unit configuration, compute the probability
        vector of the hidden units being on.

After Change

                  hidden state.
        :rtype: torch.Tensor
        
        return torch.addmm(
            self.visible_bias.data, h, self.weights.data, out=out
        ).sigmoid_()

    @auto_unsqueeze_arg(1)
    def prob_h_given_v(self, v, out=None):
        Given a visible unit configuration, compute the probability

Instances