TensorFlow - вывод только на подмножество модели

Question

Есть ли способ сделать вывод модели TF с помощью «фильтра»? Представьте, что мои тренировочные данные могут быть сгруппированы и помечены (просто пример, это не зрение ML)

1. datapoint1-> cat (родительская метка животных)
2. datapoint2-> собака (родительская метка животных)
3. datapoint3-> mazda (родительский лейбл)
4. datapoint4-> toyota (родительский лейбл)

Допустим, я тренирую это в TensorFlow. Предполагая, что при выводе модели потребитель «знает» родительскую метку - возможно ли сузить время «поиска» и вывода? Например, если клиент знает, что точка данных является автомобилем, и хотел только определить, какая это кошка, могу ли я «отфильтровать» процесс классификации TF?