For Java Spark, a solution along the lines of JavaAPISuite.mapPartitions():

rdd.mapPartitions(itr -> Collections.singletonList(Iterators.size(itr)).iterator()).collect().toString());

Just to see:

rddSliced.foreachPartition(itr -> System.out.println(Iterators.size(itr)));