tf.raw_ops.TopK

robots: noindex

Finds values and indices of the k largest elements for the last dimension.