x = self._predictions[b, i].item() - self._predictions[b, j].item()