tensorflowfeature_columntriestoreshapefeatures我正在尝试使用自定义估计器为MNIST数据集实现网络。这是我的输入函数:123456789definput_train_fn(): train,test=tf.keras.datasets.mnist.load_data() mnist_x,mnist_y=train mnist_y=tf.cast(mnist_y,tf.int32) mnist_x=tf.cast(mnist_x,tf.int32) features={'image':mnist_x} labels=mnist_y dataset=t
tensorflowfeature_columntriestoreshapefeatures我正在尝试使用自定义估计器为MNIST数据集实现网络。这是我的输入函数:123456789definput_train_fn(): train,test=tf.keras.datasets.mnist.load_data() mnist_x,mnist_y=train mnist_y=tf.cast(mnist_y,tf.int32) mnist_x=tf.cast(mnist_x,tf.int32) features={'image':mnist_x} labels=mnist_y dataset=t