(xb)
| 139 | return x - x.exp().sum(-1).log().unsqueeze(-1) |
| 140 | |
| 141 | def model(xb): |
| 142 | return log_softmax(xb @ weights + bias) |
| 143 | |
| 144 | ###################################################################################### |
| 145 | # In the above, the ``@`` stands for the matrix multiplication operation. We will call |