Now I want to do something like this:
tensor= [torch.Size([32, 1, 3, 3]), torch.Size([10]), torch.Size([128, 9216]), torch.Size([64])]
for i, t in enumerate(tensor):
    if i > = 1:
        n_t = t[i].add(t[i-1])