有没有办法在Tensorflow中找到二维张量中的前tf.nn.top_k
个值?
我可以将k
用于1-D张量,但它不能用于2-D张量。我有一个尺寸未知的二维张量,有没有办法找到最高.text {
white-space: nowrap;
}
值及其指数?
非常感谢。
答案 0 :(得分:3)
您可以在tf.nn.top_k()
之前将矩阵重塑为1-D张量,然后从1-D中计算2-D指数:
x = tf.random_uniform((3, 4))
x_shape = tf.shape(x)
k = 3
top_values, top_indices = tf.nn.top_k(tf.reshape(x, (-1,)), k)
top_indices = tf.stack(((top_indices // x_shape[1]), (top_indices % x_shape[1])), -1)
with tf.Session() as sess:
mat, val, ind = sess.run([x, top_values, top_indices])
print(mat)
# [[ 0.2154634 0.52707899 0.29711092 0.74310601]
# [ 0.61274767 0.82408106 0.27242708 0.25479805]
# [ 0.25863791 0.16790807 0.95585966 0.51889324]]
print(val)
# [ 0.95585966 0.82408106 0.74310601]
print(ind)
# [[2 2]
# [1 1]
# [0 3]]
答案 1 :(得分:0)
你能做到这一点的一种方法是重塑整个事情,比如xx= np.reshape(x,(-1,))
,然后像x[:k]
这样的事情呢?