如何理解“-input[range(target.shape[0]), target]”？

Question

在the official example of PyTorch中，给出如下损失函数。

def nll(input, target):
    return -input[range(target.shape[0]), target].mean()

loss_func = nll

如何理解上述函数中“input[range(target.shape[0]), target]”的语法？ “输入”有一个 torch.Size([64, 10])，“目标”有一个 torch.Size([64])。为什么在这里使用“range”函数？

Answer 1

范围函数用作创建从 0 到 64 的向量/列表/生成器的快捷方式。因此它本质上是 [0,1,2,...64] 的简写

要明确这一点，您可以执行以下操作：

def nll(input, target):
    minputlist = list(range(target.shape[0]))
    print(minputlist )
    return -input[minputlist, target].mean()