How to get the TopK Accuracy for the classes predictions using tf.keras.metric

Question

I have my y_true labels as [0,1,2,1,0,2] etc and I have my y_pred_probs (probabilities of each classes) as [[45,12,30,80,12,23,4