torchsummary : RuntimeError: Expected 5-dimensional input for 5-dimensional weight but...

torchsummary :
RuntimeError: Expected 5-dimensional input for 5-dimensional weight [24, 1, 3, 3, 3], but got input of size [2, 1, 1,16,256,256] instead.
Solve:

1. device = torch.device(“cuda”)#necessary for torchsummary, must to cuda
2. summary(net, input_size=(1,16,256,256))#must remove the number of N

def test():
	device = torch.device("cuda")#necessary for torchsummary, must to cuda
	net = PiaNet