loss = F.cross_entropy(y / temperature, target, reduction="mean")