<reponame>shiyuli/LibTorchDemo
import torch

torch.nn.MSELoss()              # regression
torch.nn.CrossEntropyLoss()     # classification

import torch

torch.nn.MSELoss()              # regression
torch.nn.CrossEntropyLoss()     # classification
# regression # classification