我正在使用tf.keras.Model fit_generator（https://www.tensorflow.org/api_docs/python/tf/keras/models/Model#fit_generator）将一批数据馈送到模型中。根据TensorFlow文档，fit生成器应能够接受大小2（输入，目标）或3（输入，目标，sample_weights）元组。我们的大小为2，但是类别不平衡，因此我确定了样本权重。当fit生成器返回大小为3的元组时，出现错误： ” tensorflow.python.framework.errors_impl.InvalidArgumentError：无法挤压dim [0]，预期尺寸为1，得到[batch_size]”
我正在使用tensorflow 1.12
损失函数是tf.losses.softmax_cross_entropy