While using torch.norm my code is giving nans. So, I tried changing the dtype argument  as mentioned here torch.norm — PyTorch 1.7.0 documentation 7 to dtype=torch.float64.
But it is giving the following error:
ValueError: dtype argument is not supported in frobenius norm.
As mentioned as WARNING on this page as torch.norm — PyTorch 1.7.0 documentation 7 torch.norm is deprecated and maybe removed in future release.
But the following snippet is also giving as ImportError : cannot import name 'linalg'
from torch import linalg
Please look into the issue.
Thanks.