省流版本
在输出一个张量时,你可以根据张量输出的方括号([])的数量来判断张量的维度。每个方括号对应张量的一个维度,张量的维度数就是方括号的数量。
最外面的为第一个维度,往里面依次为第2 3 ... 维度。
例如,如果你看到如下输出:
tensor([[1, 2, 3],
[4, 5, 6]])
这个张量有两个方括号,所以它是一个二维张量(矩阵)。第一个方括号包围了两个列表,表示第一个维度的大小是2。第二个方括号包围了每个列表中的元素,表示第二个维度的大小是3。
另一个示例,如果你看到如下输出:
tensor([[[1, 2],
[3, 4]],
[[5, 6],
[7, 8]]])
这个张量有三个方括号,所以它是一个三维张量。第一个方括号包围了两个元素,表示第一个维度的大小是2。第二个方括号包围了每个元素中的两个列表,表示第二个维度的大小是2。第三个方括号包围了每个列表中的元素,表示第三个维度的大小是2。
通过观察方括号的数量,你可以确定张量的维度。
标签:方括号,tensor,torch,张量,列表,维度,包围 From: https://www.cnblogs.com/hi-wind/p/17729703.html