You can set the number of slots at the ciphertext level (either during encryption or for a given ciphertext). When working multiple batch sizes (and using operations based on rotations like EvalSum, EvalSumRows, and EvalSumCols), please make sure rotation keys are generated for all indices.