We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
这里在return时不需要softmax吗?
def net(X): X = X.view((-1, num_inputs)) H = relu(torch.matmul(X, W1) + b1) return torch.matmul(H, W2) + b2