我正在尝试实施基于策略的强化学习，因此我必须在培训期间进行预测。因此，我在输入函数中将估算器置于预测模式。当我将在输入函数中创建的张量传递给预测变量的输入函数时，tensorflow抱怨
ValueError：Tensor（“ conv2d / kernel：0”，shape =（8，8，4，16），dtype = float32_ref）必须与Tensor（“ transpose：0”，shape =（1 ，84，84，4），dtype = float32，device = / device：CPU：0）。
似乎我在输入函数中创建的输入张量与预测变量使用的图不在同一张图中。在这种情况下，如何在估算器训练时进行预测？