**Describe the bug(问题描述)**  在basemodel的fit函数中,若batch_size为1,且model为多任务模型时,会导致y_pred的第0个维度被丢弃,从而产生错误。
Describe the bug(问题描述)

在basemodel的fit函数中,若batch_size为1,且model为多任务模型时,会导致y_pred的第0个维度被丢弃,从而产生错误。