import torch
x = torch.zeros((5, 2, 3, 4))
print(x.size())
print(x)
(5, 2, 3, 4)分别表示(batch size, channel, height, width)
import torch
x = torch.zeros((5, 2, 3, 4))
print(x.size())
print(x)
(5, 2, 3, 4)分别表示(batch size, channel, height, width)