Hi,
I’m afraid you won’t be able to create a single Tensor with different sizes on a single dim.
You might be able to create a Tensor with the largest size though and copy all the elements in it. Padding with 0s for the Tensors that are smaller than the full size.