top n values of a tensor to 1 others to 0

Question

I am trying to write a custom metric to a keras model for a recommendation model. I need to find a way to convert predictions vector to a vector which top 2 element in the p