y_true не может определить свою форму

Question

Я хочу извлечь часть y_true для вычисления моей собственной функции потерь, но я не смог ее выполнить.

def loss_total(self, mask):

    def loss(y_true, y_pred):

        vgg_out = []
        vgg_gt = []
        for i in range(0,9):
            y_pred_single = y_pred[:, :, :, i::9]
            y_true_single = y_true[:, :, :, i::9]


            vgg_out_single = self.vgg(y_pred_single)
            vgg_out = vgg_out.append(vgg_out_single)

            vgg_gt_single = self.vgg(y_true_single)
            vgg_gt = vgg_gt.append(vgg_gt_single)

Файл "/home/dingyuyang/.conda/envs/tensorflow/lib/python3.5/site-packages/keras/engine/base_layer.py", строка 474, в вызов output_shape = self.compute_output_shape (input_shape) Файл "/home/dingyuyang/.conda/envs/tensorflow/lib/python3.5/site-packages/keras/engine/network.py", строка 661, в compute_output_shape ключ подтверждения в layer_to_output_shapes AssertionError