如何判断torch中tensor的维度

省流版本

在输出一个张量时，你可以根据张量输出的方括号（[]）的数量来判断张量的维度。每个方括号对应张量的一个维度，张量的维度数就是方括号的数量。
最外面的为第一个维度，往里面依次为第2 3 ... 维度。

例如，如果你看到如下输出：

tensor([[1, 2, 3],
        [4, 5, 6]])

这个张量有两个方括号，所以它是一个二维张量（矩阵）。第一个方括号包围了两个列表，表示第一个维度的大小是2。第二个方括号包围了每个列表中的元素，表示第二个维度的大小是3。

另一个示例，如果你看到如下输出：

tensor([[[1, 2],
         [3, 4]],

        [[5, 6],
         [7, 8]]])

这个张量有三个方括号，所以它是一个三维张量。第一个方括号包围了两个元素，表示第一个维度的大小是2。第二个方括号包围了每个元素中的两个列表，表示第二个维度的大小是2。第三个方括号包围了每个列表中的元素，表示第三个维度的大小是2。

通过观察方括号的数量，你可以确定张量的维度。