Thanks Alwyn for your prompt reply.
Okay, We can say that  I need it in the form A=[1,8,4].
This what I had done with the loop,
x1=[]
for i in range(x.shape[1]):
x_t = x[:,i,:,:]
x1.append(x_t)
    x_t = torch.cat(x1, dim=1)