3.9节的net中是否要加入softmax #182
Description
这里在return时不需要softmax吗？
def net(X):
X = X.view((-1, num_inputs))
H = relu(torch.matmul(X, W1) + b1)
return torch.matmul(H, W2) + b2{{ message }}
这里在return时不需要softmax吗？
def net(X):
X = X.view((-1, num_inputs))
H = relu(torch.matmul(X, W1) + b1)
return torch.matmul(H, W2) + b2