code: def get_multilabel_test_input(n_train, n_test):
      """Get example multilabel inputs for attacks."""
      rng = np.random.RandomState(4)
      num_classes = max(n_train // 20, 5)  # use at least 5 classes.
      return AttackInputData(
          logits_train=rng.randn(n_train, num_classes) + 0.2,
          logits_test=rng.randn(n_test, num_classes) + 0.2,
          labels_train=get_multihot_labels_for_test(
              n_train, num_classes
          ),
          labels_test=get_multihot_labels_for_test(n_test, num_classes),
      )
  
imports: from unittest import mock
  from absl.testing import absltest
  from absl.testing import parameterized
  import numpy
  from tensorflow_privacy.privacy.privacy_tests import epsilon_lower_bound
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import membership_inference_attack
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack import models
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import AttackInputData
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import AttackType
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import DataSize
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SingleSliceSpec
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SlicingFeature
  from tensorflow_privacy.privacy.privacy_tests.membership_inference_attack.data_structures import SlicingSpec
label_desc: Use image node_id 3 for calling a global function with example usage: get_multilabel_test_input(n_train, n_test) and returns: AttackInputData