https://colab.research.google.com/drive/1N17tTaaZwZoBfSnRIXsv1uOIMH4rPpKO
"dim" 是维度dimension的缩写，比如，如果我们有一个形状为 (2, 3) 的张量，那么这个张量有两个维度。第一个维度的大小为 2，第二个维度的大小为 3。
-2维度怎么理解呢 在 PyTorch 等深度学习库中，负数在索引张量时用于表示从后往前计数。因此，-2 维度就是指倒数第二个维度。 例如，如果我们有一个四维张量，形状为 (10, 20, 30, 40)。那么： 第 0 维度的大小是 10， 第 1 维度的大小是 20， 第 2 维度的大小是 30， 第 3 维度的大小是 40。 如果我们从后往前计数： 第 -1 维度的大小是 40（也就是正数索引中的第 3 维度）， 第 -2 维度的大小是 30（也就是正数索引中的第 2 维度）。 所以，当你看到 -2 维度时，可以理解为操作是在倒数第二个维度上进行的。