7da71b22f18f5d02ca57a8e5cb5ae47338ae8a31,qucumber/rbm/purification_rbm.py,PurificationRBM,gamma_minus,#PurificationRBM#,379

Before Change

            temp += F.softplus(F.linear(v, self.weights_W, self.hidden_bias)).sum()
            temp -= F.softplus(F.linear(vp, self.weights_W, self.hidden_bias)).sum()
        else:
            temp = torch.mv(v, self.visible_bias).unsqueeze_(1) - torch.mv(
                vp, self.visible_bias
            ).unsqueeze_(0)

After Change

                F.softplus(F.linear(v, self.weights_W, self.hidden_bias)).sum(1)
            )

            temp2 = torch.mv(vp, self.visible_bias) + (
                F.softplus(F.linear(vp, self.weights_W, self.hidden_bias)).sum(1)
            )

            if expand:
                temp = temp1.unsqueeze_(1) - temp2.unsqueeze_(0)
            else:
                temp = temp1 - temp2

        return 0.5 * temp

Instances