I have the following code, written in tf.keras with Tensorflow 2. Basically; I need the cross entropy term\'s gradient with respect to the variable self.temperature
self.temperature