Issues in Integrating Differential Privacy in Keras Custom Loss Function

[Jeet0204]

How to return a vector loss in Keras Custom Loss function, So that I don't have a issue with the following error: ValueError: Dimension size must be evenly divisible by 256 but is 1 for 'Reshape' (op: 'Reshape') with input shapes: [], [2] and with input tensors computed as partial shapes: input[1] = [256,?].