Using torch::linalg::linalg_norm() like torch::linalg::linalg_norm(tensor, 2, 2) gives this error
error: no matching function for call to ‘linalg_norm(at::Tensor&, int, int)’
I want to do this

np.linalg.norm(diff, ord=2, axis=2)

in Libtorch.