... y = F.relu(self.bn1(self.linear1(input))) y = F.softmax(self.linear2(y), dim=1) return y model = network() x = torch.randn(10, 40) output = model(x)
確定! 回上一頁