簡體   English   中英

如何在張量流中獲取矩陣中每一行的最大值和次級最大值?

[英]How to get the maximum and secondary maximum values of each row in a matrix in tensorflow?

如何在張量流中獲取矩陣中每一行的最大值和次級最大值? 例如,我的矩陣張量為[[1,2,3],[6,5,4],[7,9,8]。 然后,我想獲得每行的最大值矢量,例如[3,6,9]和次級值矢量[2,4,8]。 可能的問題可以分為兩部分:1.查找值索引2.從矩陣中收集值並形成一個新向量。

您可能想使用tf.nn.top_k來完成您想要的工作:

查找最后一個維度的k個最大條目的值索引。 ...對於矩陣(分別為較高秩的輸入),計算每行中的前k個條目(沿最后一維的對應向量)

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM