擅长:python、mysql、java
<p>我们可以修改BlueSun的解决方案,使其在GPU上更快:</p>
<pre><code>def get_median(v):
v = tf.reshape(v, [-1])
m = v.get_shape()[0]//2
return tf.reduce_min(tf.nn.top_k(v, m, sorted=False).values)
</code></pre>
<p>这与(以我的经验)使用<code>tf.contrib.distributions.percentile(v, 50.0)</code>一样快,并返回一个实际元素。</p>