如何在张量流中获得矩阵中每行的最大值和次要最大值？例如 . 我有一个矩阵张量为[[1,2,3]，[6,5,4]，[7,9,8]] . 然后我想获得每行的最大值向量，如[3,6,9]和次要值向量[2,4,8] . 可能是问题可以分为两部分：1 . 找到值索引2.从矩阵中收集值并形成一个新的向量 .
您可能希望使用tf.nn.top_k，它正是您正在寻找的：
查找最后一个维度的k个最大条目的值和索引 . ...对于矩阵（或更高级别的输入），计算每行中的前k个条目（沿最后一个维度分别为向量）
1 回答
您可能希望使用tf.nn.top_k，它正是您正在寻找的：