deepfold.utils.dist_utilsΒΆ Functions get_pad_size(tensor, dim, num_chunks) pad_tensor(tensor, dim, pad_size)