[checkpoint_io] gather tensor before unpad it if the tensor is both padded and distributed

[checkpoint_io] gather tensor before unpad it if the tensor is both padded and distributed #5846