tf.keras.backend.batch_set_value