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

[checkpointio] gather tensor before unpad it if the tensor is both padded and distributed #5856