通过示例讲解查看张量维度的方法,包括整体维度和某一维的数值。
随机生成一个张量数组:
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
查看第一维和第二维大小的方式与此类似。
通过示例讲解查看张量维度的方法,包括整体维度和某一维的数值。
随机生成一个张量数组:
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
查看第一维和第二维大小的方式与此类似。