Node: 'mean_squared_error/SquaredDifference' required broadcastable shapes [[{{node mean_squared_error/SquaredDifference}}]] [Op:__inference_train_function_2983] 2023-05-22 22:06:16.671412: I tensorflow/stream_executor/cuda/cuda_dnn.cc:368] Loaded cuDNN version 8901 2023-05-22 22:06:18.052867: W tensorflow/core/framework/op_kernel.cc:1733] INVALID_ARGUMENT: required broadcastable shapes根据这个参数应该修改哪个参数

根据错误提示，需要修改的参数是输入数据的形状和大小，以便使它们可以广播到相同的形状。您可以检查您的输入数据和模型的输入形状是否匹配，并尝试调整它们以满足广播要求。您还可以尝试使用`tf.expand_dims()`函数来增加维度，以便使它们可以广播到相同的形状。另外，您还可以在模型中添加一个Reshape层或Flatten层来调整输入数据的形状。