一些小小小细节20220217

logits.argmax(dim=1)

emmmm好像是一句非常简单的代码 ，但是我完全看不懂。。。。
这个常用在最后的预测值上，
be like：

# Forward
logits = model(g, features)

# Compute prediction
pred = logits.argmax(1)

经过一个argmax的操作，dim=1 意味着找到张量中各自的最大值所在索引。也就是找到每个样本的全连接输出中最大的那一个。 最有可能的预测值。
直接上例子：

A = torch.tensor([[0,1], [0,2], [