把dim理解成“降维打击”，你指定哪一维，就降那一维。

    dim默认是None，也就是把所有dim都”全部降维”，最后只会有一个值
    如果指定一个数字，那就是指定的那个dim会被”降维”成一个数值

    比如一个shape=[2, 3, 4]，dim=1，结果的shape就是[2, 4]
    也就是[2, *, 4]不动，但是dim=1的每 “3个数值的向量” 会通过运算符 变成一个数字

    dim是可以写多个值的
    image.png
    argmax运算也是同理，但是这个呢，又比较特别，要么不输dim，要么只能指定一个维度，哈哈。。。

    logits.ndim == 2
    dim0是batch_size
    dim1是神经网络输出的神经元数
    要计算的是dim=1中的每个最大值在第几个神经元，就是每个样本的类别