Pytorch concatenate scalars. stack but this requires the dimen… torch.