Как получить верхние значения k и индекс слоя Keras? - PullRequest
0 голосов
/ 08 июля 2019

Я использую модель Keras, в которой мне нужно получить верхние значения k и индекс слоев Keras как два слоя 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)

...