import torch import torch.nn as nn class SimpleModel(nn.Module): def __init__(self, input_size, ... self.softmax = nn.Softmax(dim=1)
確定! 回上一頁