Missed TensorFlow Dev Summit? Check out the video playlist. Watch recordings


View source on GitHub

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

    dataset, label_key, desired_label

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


  • dataset: the base tf.data.Dataset 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 tf.data.Dataset that is composed of only examples that have a label matching desired_label.