torch.size()的理解

import torch

x = torch.zeros((5, 2, 3, 4))
print(x.size())
print(x)

（5, 2, 3, 4）分别表示（batch size, channel, height, width）