张量形状为(batch_size, 13, 13, 3, 1)，其中最后一个维度是客观得分。如何仅保持top_k客观性得分而不弄乱张量（如果不在top-k中，则在最后一个维度中将所有内容设为零）。
谢谢
答案 0 :(得分：0)
您可以使用tf.math.top_k
tf.math.top_k(tensor, k= top_k_value)
要将最后一个维度的所有内容都设为0，可以考虑使用以下内容：
tf.mul(tensor, tf.constant([0]))
对于后者，您将最终使张量的所有元素均为0