pytorch：Tensor维度初步理解

[1, 2, 3]

这是一维向量，size=3

---

[
    [1, 2, 3]
                ]

这是二维向量，size=1*3

---

[

    [
        [1, 2],
        [3, 4],
        [5, 6]
                 ],

    [
        [1, 2],
        [3, 4],
        [5, 6]
                 ]

                     ]

这是三维向量，size=2*3*2

---

结论：pytorch中的tensor维度可以通过第一个数前面的中括号数量来判断，有几个中括号维度就是多少。拿到一个维度很高的向量，将最外层的中括号去掉，数最外层逗号的个数，逗号个数加一就是最高维度的维数，如此循环，直到全部解析完毕。