Function log_softmax

beginner_source/nn_tutorial.py:138–139

(x)

# automatically.

def log_softmax(x):
    return x - x.exp().sum(-1).log().unsqueeze(-1)

def model(xb):
    return log_softmax(xb @ weights + bias)