[Task] Add topK layer in model to return topK items at inference step

### Description

At inference, we generate scores for all the items in the catalog per session. If we add a topK layer, we will be able to return topK items with the highest scores at inference step automatically.