TensorFlow 2.4 behaviour with custom loss functions: Cannot convert a symbolic Keras input/output to a numpy array

Question

I\'m learning TensorFlow pretty much from scratch, and finding it hard to replicate examples on latest versions of TF.

In a typical VAE example of MNIST I have a segm