【Python】查看张量（tensor）数据维度和每一维大小示例

通过示例讲解查看张量维度的方法，包括整体维度和某一维的数值。

随机生成一个张量数组：

import torch
a=torch.rand(1,3,480,640)

然后使用.shape查看数组维度，
输入：

x=a.shape
print(x)

输出：

torch.Size([1, 3, 480, 640])

查看第3未的数值：

x_3=a.shape[3]

输出：

640

查看第一维和第二维大小的方式与此类似。