The saved dataset is saved in various file "shards". By default, the dataset output is divided to shards in the spherical-robin vogue but tailor made sharding can be specified via the shard_func function. As an example, you can save the dataset to using a single shard as follows:This probabilistic interpretation subsequently can take a similar vari