Пользовательская функция потерь Keras: пересечение

Question

Я хочу разработать специальную функцию потерь Keras, y_pred которой равен 2 тензору, в дальнейшем я буду называть этот тензор A и B.

Где формы A и B (размер партии, временной шаг, вектор)

Моя функция потерь - это abs (sum (diff (A, correct (A, B))) - sum (diff (B, correct (A, B))))

    A = [[1.1, 1.2, 1.3],[2.1, 2.2, 2.3], [3.1, 3.2, 3.3], [4.1, 4.2, 4.3], [7.1, 7.2, 7.3]]
    B = [[1.1, 1.2, 1.3],[5.1, 5.2, 5.3], [6.1, 6.2, 6.3], [3.1, 3.2, 3.3], [4.1, 4.2, 4.3]]
    C = intersect(A, B)
        # set operation
        # now C should be like [[1.1, 1.2, 1.3], [3.1, 3.2, 3.3], [4.1, 4.2, 4.3]]
        # because it's in A and B
    D = diff(A, C)
        # set operation A-C
        # then D should be like  [[2.1, 2.2, 2.3], [7.1, 7.2, 7.3]]]

Для sum () и abs () работают как reduce_sum и обычная функция abs

Итак, как я могу вычислить такую ​​функцию потерь?