nn里面的都是类,需要先实例化再调用,其实就是需要放入模型的层中使用

  1. from torch import nn
  2. loss_fun = nn.CrossEntropyLoss()
  3. loss = loss_fun(X,y)

F里面的是函数(nn的对应层的forward中会调用对应的F函数),可以直接使用

  1. import torch.nn.functional as F
  2. loss = F.cross_entropy(X, y)