如何在TensorFlow中獲得最大3張張？

Question

我在seq2seq基於TensorFlow建立自己的關注模型。我想知道我怎麼可以這樣做如下的僞代碼TensorFlow：

att = [0.2,0.3,0.5,0.1,0.9,0.2]# This is a tensor or a list of tensors 
result = maximum3numbers(att) 

接受的結果可能是：

result = [0.9,0.5,0.3] 

或

result = [4,2,1] 

Answer 1

我認爲top_k運營商應該做的你需要什麼：

https://www.tensorflow.org/versions/r0.9/api_docs/python/nn.html#top_k