Nikronic:

But typically, you can calculate it using torch.dist(output, input, 2) . Here is the documentation.


I think torch.dist(ne, naa, 2) will do the trick. No need to convert it to numpy or transfer it to CPU.