小编use*_*134的帖子

有没有办法k在 Tensorflow 的二维张量中找到最高值？

我可以tf.nn.top_k用于一维张量，但它不能用于二维张量。我有一个大小未知的二维张量，有没有办法找到最高k值及其索引？

非常感谢。