如何获得Keras层的前k个值和索引？

Question

我正在使用Keras模型，在该模型中，我需要获取k层的k值和索引作为两个Keras层。有什么办法可以做到这一点。

在Tensorflow中，我们有tf.nn.top_k。但是我没有找到Keras.Backend.top_k

out_vector = Dense(self.vocab_size, activation='softmax')(mlp_input)

values, index = Lambda(lambda x:K.top_k(x, 5))(out_vector)