Build a new dataset that only yields examples with a particular label.

This can be used for creating pathological non-iid (in label space) datasets.

dataset The base that yields examples that are structures of string key -> tensor value pairs.
label_key The str key that holds the label for the example.
desired_label The label value to restrict the resulting dataset to.

A that is composed of only examples that have a label matching desired_label.