torch tag

I am creating a deep learning image classification model in Pythorch.

When I tried torch.sqrt() as part of the loss function of the model, I encountered a problem where nan occurs when backing up.The reason seems to be that the vector element entered in torch.sqrt() is ...