code: def test_multi_1x1_invertibility(self, func, multihead_split):
      name = "multi_1x1"
      x, x_mask, _ = self.get_data()
  
      x_inv, logabsdet = func(
          name, x, x_mask, multihead_split, inverse=False, dtype=DTYPE
      )
      x_inv_inv, logabsdet_inv = func(
          name,
          x_inv,
          x_mask,
          multihead_split,
          inverse=True,
          dtype=DTYPE,
      )
      self.evaluate(tf.global_variables_initializer())
      (
          x,
          x_mask,
          x_inv,
          x_inv_inv,
          logabsdet,
          logabsdet_inv,
      ) = self.evaluate(
          [x, x_mask, x_inv, x_inv_inv, logabsdet, logabsdet_inv]
      )
      diff = x - x_inv_inv
      logabsdet_sum = logabsdet + logabsdet_inv
      logabsdet_ = logabsdet / np.sum(x_mask, -1)
      self.assertTrue(np.allclose(diff, 0.0, atol=1e-5))
      self.assertTrue(np.allclose(logabsdet_, 0.0, atol=1e-5))
      self.assertTrue(np.allclose(logabsdet_sum, 0.0, atol=1e-5))
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import os
  import tempfile
  from absl.testing import parameterized
  import numpy
  from tensor2tensor.layers import common_attention
  from tensor2tensor.layers import transformer_glow_layers
  from tensor2tensor.layers import transformer_glow_layers_ops
  from tensor2tensor.models import transformer
  import tensorflow.compat.v1
label_desc: Use image node_id 6 for calling the TransformerGlowLayersTest obj's underlying member method code with example usage: obj.test_multi_1x1_invertibility(func, multihead_split) without return types