Currently, I\'m using tf.keras(Tensorflow 2.4).
And, i\'m implement \'mask-image gradient consistency loss\' in Keras. But, i don\'t know that how to access training inpu