import torch.nn as nn import torch.nn.functional as F class Model(nn.Module): ... >>> loss = nn.MSELoss() >>> input = torch.randn(3, 5, requires_grad=True) ...
確定! 回上一頁