Hello everyone.
I’m having problems with torch.stack(), I reported a small example with the error that I get
import torch
a = torch.zeros(3,2,1,2)
b = torch.zeros(2,2,1,2)
torch.stack([a,b], dim=0)

RuntimeError: invalid argument 0: Sizes of tensors must match except in dimension 0. Got 3 and 2 in dimension 1 at /opt/conda/conda-bld/pytorch_1550780889552/work/aten/src/TH/generic/THTensorMoreMath.cpp:1307

why do I have this error message?