将Keras自定义损失函数中的张量拟合到另一个网络

Question

我正在def loss(y_true, y_pred)的Keras中开发自定义损失函数。 但是我想在另一个神经网络中重用其中一个张量（y_pred）。我将其传递给Model对象。

当我尝试运行代码时，出现错误： ValueError: When feeding symbolic tensors to a model, we expect the tensors to have a static batch size. Got tensor with shape: (None, 256, 256, 3)。

我编译模型： network.compile(loss=loss, optimizer='sgd')

并使用network.fit(x=random_array, y=np.zeros((BATCH_SIZE, imgdim[0], imgdim[1], imgdim[2])))

运行这些损失函数

如何使y_pred张量适合另一个网络？