查找张量不同维度的大小

查找张量不同维度的大小

代码：

import torch

x = torch.ones([4, 3, 160, 160])

print(x.shape[0])

运行结果：