import os import torch import torch.nn as nn import torch.nn.functional as F ... probs = torch.exp(cnn_model(x)) # b x 10 scores = score(torch.arange(10, ...
確定! 回上一頁