我知道计算逆的实现(即 tf.nn.top_k 计算沿给定轴的 K 个最大值),但我找不到找到 K 最小值的内置方法 值(value)观。
除了取逆或玩弄集合之外,是否可以使用 tensorflow 库执行此操作,或者我必须自己设计一些东西吗?
干杯/
最佳答案
不,没有您可以使用的单一功能。使用tf.nn.top_k()
没有问题以此目的。只是否定论点:
-tf.nn.top_k(-A)
将执行与 tf.negative(tf.nn.top_k(tf.negative(A)))
相同的操作
关于python - 张量的最小 K 值?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/44548227/