在自定义损失函数中重塑张量

在自定义损失函数中重塑张量