How can I vectorize the operation where all non zero values in a PyTorch Tensor are copied to the left most position without any gaps?

Question

I have a PyTorch tensor with 0 and non 0 values, and I want to copy the non zero values in a new tensor such that it is as compact as possible. ie its shape is [num_rows,max([nu