Here is my code which I try to write the loss function and then added to my VAE but this error happened. I found something similar but there wasn\'t any good solution for this i