when i tried
input = input.view(input.size(0),  -1)
which outputs
torch.Size([64, 150528])
before
input=input.permute(1,2,0,3,4).contiguous()
i had the follwing error
input=input.permute(1,2,0,3,4).contiguous()
RuntimeError: number of dims don’t match in permute
as input is (64,3,224,244)
and i need to expand it as i have to match with (64,3,1,7,7) kernal