63352897d1db6a7a009a1097ee9b16be13fc53b3,qucumber/rbm/binary_rbm.py,BinaryRBM,effective_energy_gradient,#BinaryRBM#,102

Before Change

            hb_grad = -prob
        else:
            W_grad = -torch.einsum("ij,ik->jk", (prob, v))
            vb_grad = -torch.einsum("ij->j", (v,))
            hb_grad = -torch.einsum("ij->j", (prob,))

        return parameters_to_vector([W_grad, vb_grad, hb_grad])

After Change

            hb_grad = -prob
        else:
            W_grad = -torch.matmul(prob.t(), v)
            vb_grad = -torch.sum(v, 0)
            hb_grad = -torch.sum(prob, 0)

        return parameters_to_vector([W_grad, vb_grad, hb_grad])

Instances