【发布时间】：2018-07-16 12:03:18
【问题描述】：
如何在tensorflow中获取矩阵中每一行的最大值和次最大值？例如。我有一个矩阵张量为 [[1,2,3],[6,5,4],[7,9,8]]。然后我想得到每一行的最大值向量，如[3,6,9]和二级值向量[2,4,8]。可能问题可以分为两部分：1.找到值索引 2.从矩阵中收集值并形成一个新的向量。
【问题讨论】：
标签： python tensorflow
【发布时间】：2018-07-16 12:03:18
【问题描述】：
您可能想使用tf.nn.top_k，它完全符合您的要求：
查找最后一个维度的 k 个最大条目的值 和 索引。 ... 对于矩阵（分别是更高级别的输入），计算每行中的前 k 个条目（分别是最后一个维度的向量）
【讨论】：
-
是的！这正是我想要的！非常感谢！