y_hat

y = torch.tensor([0, 2]) y_hat = torch.tensor([[0.1, 0.3, 0.6], [0.3, 0.2, 0.5]]) y_hat[[0, 1], y] 这段代码使用了两个 PyTorch 张量 y 和 y_hat，并通过 y_hat[[0, 1], y] ......