Performance issue in the definition of build_loss, model.py(P1)

[DLPerf]

Hello, I found a performance issue in the definition of build_loss, model.py, tf.expand_dims(target_image, axis=-1) will be created repeatedly during program execution, resulting in reduced efficiency. I think it should be created before the loop.

Looking forward to your reply. Btw, I am very glad to create a PR to fix it if you are too busy.

[shaohua0116]

Thanks for the catch! Please create a PR so that your effort is recognized.