Every convnet classifier does that, unless I'm misunderstanding your question.

→ View original post on X — @jeremyphoward