Когда я использую такие функции, как tf.math.top_k () или tfa.image.median_filter2d (), я не понимаю, как можно рассчитать градиент для такой функции.