code: def test_get_logits(self):
      import tensorflow as tf
  
      model = KerasModelWrapper(self.model)
      x = tf.placeholder(tf.float32, shape=(None, 100))
      preds = model.get_probs(x)
      logits = model.get_logits(x)
  
      x_val = np.random.rand(2, 100)
      tf.global_variables_initializer().run(session=self.sess)
      p_val, logits = self.sess.run(
          [preds, logits], feed_dict={x: x_val}
      )
      p_gt = np.exp(logits) / np.sum(
          np.exp(logits), axis=1, keepdims=True
      )
      self.assertTrue(np.allclose(p_val, p_gt, atol=1e-6))
  
file_path: cleverhans/cleverhans_v3.1.0/tests_tf/test_utils_keras.py
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from __future__ import unicode_literals
  import unittest
  import numpy
  from tensorflow import keras
  from cleverhans.utils_keras import KerasModelWrapper
label_desc: Use image node_id 4 for calling the TestKerasModelWrapper obj's underlying member method code with example usage: obj.test_get_logits() without return types