Skip to main content
 首页 » 编程设计

python之Tensorflow in_top_kv

2024年12月31日33落叶无声

您好,我正在尝试构建一个具有 11 个输入和 2 个输出的简单 rnn X=tf.placeholder(tf.float32,[无,n_steps,n_inputs]) y=tf.placeholder(tf.int32,[None,n_steps,n_outputs])

我知道 rnn 除了 [batch_size,n_steps,n_inputs] 形状的输入,所以这就是为什么我将占位符塑造成这样的原因

但是当我运行代码时,我得到一个错误

ValueError: Shape must be rank 2 but is rank 3 for 'in_top_k/InTopKV2' (op: 'InTopKV2') with input shapes: [1,270,2], [1,270,2], [].

错误似乎起源于此:correct = tf.nn.in_top_k(logits,tf.reshape(y,[1,n_steps,n_outputs]),1)

我尝试过 reshape logits、压缩 logits、扩大 y 维度,但似乎没有任何效果。

我注意到的一个区别是,当我用

压缩 logits 时
tf.squeeze(logits) 

现在的错误是

ValueError: Shape must be rank 1 but is rank 3

这是我能够取得的唯一“进步”,我们将不胜感激。

请放轻松,这是我的第一个问题

请您参考如下方法:

您必须将输入重新整形为 2D 张量,然后您可以将结果重新整形为所需的形状:

logits_res = tf.reshape(logits, (-1, n_outputs)) 
y_res = tf.reshape(y, (-1, n_outputs)) 
correct_res = tf.nn.in_top_k(logits_res, y_res, 1) 
correct = tf.reshape(correct_res, (-1, n_steps))