如何求张量/矩阵中列正元素的中位数？

Question

特别是给定一个二维矩阵，如何为每一列的正元素求中值？你知道吗

从数学上讲：返回B，其中B[i] = median({A[j, i] | A[j, i] > 0})

我知道中位数可以用tf.contrib.distributions.percentile来计算

tf.boolean_mask(A, tf.greater(A, 0))输出一维列表而不是矩阵。你知道吗

Answer 1

你可以在列切片上循环并像这样过滤。你知道吗

inputlist = [[5 , -10 ] ,
           [10 , 3 ] ,
           [15 , -5 ]]

x = tf.Variable(initial_value=inputlist)

sess = tf.Session()
sess.run(tf.global_variables_initializer())

for i in range(x.get_shape().as_list()[1]) : #loop over columns
    print( sess.run(tf.contrib.distributions.percentile(tf.gather(x[:,i],
                                                                     tf.where(tf.greater(x[:,i],
                                                                                         0))),
                                                        50.0)))