Keras：如何在自定义损失中获取y_true尺寸？

Question

我试图用喀拉拉河写我的自定义损失

假设y_true为图片且形状= [B，H，W，C]

if i != (n-1) && j != (n-1) then ()
else (* do the rest *)

但是我得到一个错误：

def myloss(y_true, y_pred, p=10000):

    mask = y_true[0]
    mask_res = mask[:, :, 0] + mask[:, :, 1] * 256 + mask[:, :, 2] * 256 * 256
    mask_res = tf.reshape(mask_res, [mask_res.shape[0] * mask_res.shape[1]])
    ...

TypeError：无法将类型的对象转换为Tensor。内容：[Dimension（None）]。考虑将元素强制转换为受支持的类型。

我如何获得形状y_true？