时间戳

首页 > 解决方案 > 为什么 TensorFlow 会提示我将错误的形状和类型输入到占位符中？

python - 为什么 TensorFlow 会提示我将错误的形状和类型输入到占位符中？

问题描述

我想不通。我一直在来回走动，我知道我可以复制和粘贴一个工作教程，但我想了解为什么这不起作用。

import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.examples.tutorials.mnist import input_data


mnist = input_data.read_data_sets('MNIST_data', one_hot=True)   

#simple constants
learning_rate = .01
batch_size = 100
training_epoch = 10
t = 0
l = t

#gather the data
x_train = mnist.train.images
y_train = mnist.train.labels
batch_count = int(len(x_train)/batch_size)

#Set the variables
Y_ = tf.placeholder(tf.float32, [None,10], name = 'Labels')
X = tf.placeholder(tf.float32,[None,784], name = 'Inputs')
W = tf.Variable(tf.zeros([784,10]))
b = tf.Variable(tf.zeros([10]))

#Build the graph (Y = WX + b)
Y = tf.nn.softmax(tf.matmul(X,W) + b, name = 'softmax')

cross_entropy = -tf.reduce_mean(Y_ * tf.log(Y)) * 1000.0

correct_prediction = tf.equal(tf.argmax(Y,1), tf.argmax(Y_,1))

accuracy = tf.reduce_mean(tf.cast(correct_prediction, tf.float32))

train_step = tf.train.GradientDescentOptimizer(learning_rate).minimize(cross_entropy) 

with tf.Session() as sess:
    sess.run(tf.global_variables_initializer())


    for epoch in range(training_epoch):
        for i in range(batch_count):
            t += batch_size
            print(y_train[l:t].shape)
            print(x_train[l:t].shape)
            print(y_train[l:t].dtype)
            sess.run(train_step,feed_dict={X: x_train[l:t], Y: y_train[l:t]})
            l = t
        print('Epoch = ', epoch)
    print("Accuracy: ", accuracy.eval(feed_dict={X: x_test, Y_: y_test})) 
    print('Done')

错误信息：

InvalidArgumentError: You must feed a value for placeholder tensor 'Labels_2' with dtype float and shape [?,10]
     [[Node: Labels_2 = Placeholder[dtype=DT_FLOAT, shape=[?,10], _device="/job:localhost/replica:0/task:0/device:GPU:0"]

我也明白，要让它发挥作用，我还需要添加更多内容，但我现在想自己努力解决这个问题。我在 jupyter 笔记本上运行它。我很肯定它y_train有一个形状 (100, 10) 和一个 float64 类型。

我已经被困了几天，所以我很感激帮助。

标签： pythontensorflowmachine-learningmnisttensor

解决方案

您需要在Y_调用时输入占位符张量sess.run。

在feed_dict，只需更改Y: y_train[l:t]为Y_: y_train[l:t]。这将y_train[l:t]输入占位符。

推荐阅读