形狀必須相等

Question

我想從預先訓練的模型中進行轉移學習。 我正在遵循Tensorflow的再培訓指南 。

但是，我陷入了一個錯誤tensorflow.python.framework.errors_impl.InvalidArgumentError: Shapes must be equal rank, but are 3 and 2 for 'input_1/BottleneckInputPlaceholder' (op: 'PlaceholderWithDefault') with input shapes: [1,?,128].

# Last layer of pre-trained model 
# `[<tf.Tensor 'embeddings:0' shape=(?, 128) dtype=float32>]`

with tf.name_scope('input'):
    bottleneck_input = tf.placeholder_with_default(
        bottleneck_tensor,
        shape=[None, 128],
        name='BottleneckInputPlaceholder')

有任何想法嗎？

Answer 1

發生這種情況是因為您的bottleneck_tensor的形狀為[1, ?, 128]並且您明確聲明該形狀應為[?, 128] 。 您可以使用tf.squeeze減少將張量轉換為所需形狀

tf.squeeze(bottleneck_tensor)