code: def test_multi_iter_gen_adv_neighbor_for_feature_columns_with_int_feature(
      self,
  ):
      (
          x,
          y,
          w,
          expected_neighbor_fc1,
          expected_neighbor_fc2,
      ) = (
          self._test_multi_iter_gen_adv_neighbor_for_feature_columns_with_int_feature()
      )
  
      # Simple linear regression.
      x_stacked = tf.concat(
          [x["fc1"], x["fc2"], tf.cast(x["fc_int"], tf.dtypes.float32)],
          axis=1,
      )
      y_hat = tf.matmul(x_stacked, w)
      loss = tf.math.squared_difference(y, y_hat)
      loss_fn = tf.math.squared_difference
  
      @tf.function
      def model_fn(inp):
          x_stacked = tf.concat(
              [
                  inp["fc1"],
                  inp["fc2"],
                  tf.cast(inp["fc_int"], tf.dtypes.float32),
              ],
              axis=1,
          )
          return tf.matmul(x_stacked, w)
  
      adv_config = configs.AdvNeighborConfig(
          feature_mask=None,
          adv_step_size=0.1,
          adv_grad_norm="l2",
          pgd_iterations=2,
          pgd_epsilon=0.13,
      )
      adv_neighbor, _ = adv_lib.gen_adv_neighbor(
          x,
          loss,
          adv_config,
          pgd_model_fn=model_fn,
          pgd_loss_fn=loss_fn,
          pgd_labels=y,
      )
  
      actual_neighbor = self.evaluate(adv_neighbor)
      self.assertAllClose(expected_neighbor_fc1, actual_neighbor["fc1"])
      self.assertAllClose(expected_neighbor_fc2, actual_neighbor["fc2"])
      self.assertAllClose([[2]], actual_neighbor["fc_int"])
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import neural_structured_learning.configs
  from neural_structured_learning.lib import adversarial_neighbor
  import numpy
  import tensorflow
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 21 for calling the GenAdvNeighborTest obj's underlying member method code with example usage: obj.test_multi_iter_gen_adv_neighbor_for_feature_columns_with_int_feature() and returns: tf