如何使用尺寸为无的tf.nn.top_k

Question

我想使用tf.nn.top k替换argsort（numpy）
但似乎tf.nn.top_k不接受None维度 这是我的代码

cond1 = tf.greater_equal(ws, min_size) # assume shape is (100,)
cond2 = tf.greater_equal(hs, min_size) # assume shape is (100,)
cond = cond1 & cond2 # shape is (100)

# cause I don't give x and y, so tf.where return index of True element
# but number of True is unknow now
keep = tf.where(cond) # so shape is (?,1)
keep = tf.reshape(keep, [-1]) # shape is (?,)

val = tf.gather(val, keep) # shpae is (?,)
argsort = tf.nn.top_k(val, val.get_shape()[0]) 
# ValueError: Cannot convert an unknown Dimension to a Tensor: ?

Answer 1

从here找到答案 维度仍然没有，但它的工作，惊讶

{{1}}