Question

我有一个这样的张量：

sim_topics = [[0.65 0.   0.   0.   0.42  0.   0.   0.51 0.   0.34 0.]
              [0.   0.51 0.   0.  0.52  0.   0.   0.   0.53 0.42 0.]
              [0.    0.32 0.  0.50 0.34  0.   0.   0.39 0.32 0.52 0.]
              [0.    0.23 0.37 0.   0.    0.37 0.37 0.   0.47 0.39 0.3 ]]

和这样的一个布尔张量：

bool_t = [False  True  True  True]

我想基于sim_topics中的bool标志选择bool_t的一部分，其方式是每行选择top k smallest个值（如果该行为true，则不保留为是）。

所以预期的输出将是这样的：（在k=2）

[[0.65 0.   0.    0.   0.42  0.   0.   0.51 0.   0.34 0.]
 [0.   0.51 0.   0.  0.52  0.   0.   0.   0.53 0.42 0.]
 [0.   0.32 0.    0.50 0    0    0.   0.   0    0.32 0 ]
 [0.   0.23 0     0.   0.    0    0    0.   0    0    0.3 ]]

我试图先通过使用boolean_mask和where来获得所需的索引，然后再去获得最小的索引来完成此任务。但是，当我使用where时，它没有给我索引zero所在的地方。

Answer 1

k = 2
dim0 = sim_topics.shape[0]
a = tf.cast(tf.equal(sim_topics,0), sim_topics.dtype)
b = tf.reshape(tf.reduce_sum(a,1) + k, (dim0,-1))
c = tf.cast(tf.argsort(tf.argsort(sim_topics,1),1), sim_topics.dtype)
d = tf.logical_or(tf.less(c,b),tf.reshape(tf.logical_not(bool_t),(dim0,-1)))
with tf.Session() as sess:
  print(sess.run(sim_topics * tf.cast(d,sim_topics.dtype)))


[[0.65 0.   0.   0.   0.42 0.   0.   0.51 0.   0.34 0.  ]
 [0.   0.51 0.   0.   0.   0.   0.   0.   0.   0.42 0.  ]
 [0.   0.32 0.   0.   0.   0.   0.   0.   0.32 0.   0.  ]
 [0.   0.23 0.   0.   0.   0.   0.   0.   0.   0.   0.3 ]]

我如何基于张量流中的条件获得最高的最小张量值

1 个答案: