albanD:

torch.cat(your_list, 0)


Thanks. Tried torch.cat(my_list, 0), but got this error
RuntimeError: zero-dimensional tensor (at position 0) cannot be concatenated

@smth solution works
Thank you all for the quick replies