无法挤压dim [1]，预期尺寸为1，得到2

您的标签形状错误。请参阅文档：

使用

sparse_categorical_crossentropy

损失时，您的目标应该是 整数
目标。如果您有明确的目标，则应使用

categorical_crossentropy

因此，您需要将标签转换为整数：

train_labels = np.argmax(train_labels, axis=1)