Kasper Yang
Results
1
issues of
Kasper Yang
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)