如何判断torch中tensor的维度

发布时间 2023-09-26 11:24:58作者: 诗子黎

省流版本

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

例如,如果你看到如下输出:

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

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

另一个示例,如果你看到如下输出:

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

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

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

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