code: def _compare_expand_tensor_with_kronecker_product(
      self, tensor, block_dim
  ):
      with self.cached_session() as session:
          compat.initialize_variables(self)
          expanded_tensor = pruning_utils.expand_tensor(
              tensor, block_dim
          )
          kronecker_product = pruning_utils.kronecker_product(
              tensor, tf.ones(block_dim)
          )
          expanded_tensor_val, kronecker_product_val = session.run(
              [expanded_tensor, kronecker_product]
          )
          self.assertAllEqual(
              expanded_tensor_val, kronecker_product_val
          )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  from absl.testing import parameterized
  import tensorflow
  from tensorflow_model_optimization.python.core.keras import compat
  from tensorflow_model_optimization.python.core.sparsity.keras import pruning_utils
label_desc: Use image node_id 2 for calling the PruningUtilsParameterizedTest obj's underlying member method code with example usage: obj._compare_expand_tensor_with_kronecker_product(tensor, block_dim) without return types