kervolution-pytorch
kervolution-pytorch copied to clipboard
Little error on Gaussian
The code should be following:
l = x_unf[:, :, :, None].transpose(1, 3) - w.view(w.size(0), 1, 1, -1).transpose(0,1) l = torch.sum(l**2, 3)