Hi albanD,
Thanks for the response.
The size of a and b are this:
print(a.size(), b.size())



torch.Size([5]) torch.Size([5, 3])



How would you apply torch.stack? I’ve tried dim = -2, -2, 0, 1, but seem to get errors.
inconsistent tensor sizes at /opt/conda/conda-bld/pytorch_1512387374934/work/torch/lib/TH/generic/THTensorMath.c:2864