Only using specific features from TFRecord in model

Question

I\'m trying to figure out how I can restrict the features used within my DNN, when I\'m streaming data to my model using TFrecords and the Tensorflow Data module. I\'m guess