在tensorflow中沿着(batch，13，13，3，1)张量的最后一维的top k

Question

张量形状是(batch_size, 13, 13, 3, 1)，其中最后一个维度是客观性分数。如何在不摆弄张量的情况下只保留top_k对象分数(如果不在top-k中，则使最后一维中的所有内容都为零)。

谢谢

Answer 1

您可以使用tf.math.top_k

tf.math.top_k(tensor, k= top_k_value)

要将最后一个维度中的所有内容都设置为0，可以考虑使用以下方法：

tf.mul(tensor, tf.constant([0]))

使用后一种方法，最终张量的所有元素都为0