code: def test_pack_sequences_k_multi_input(self):
      input_tokens = tf.convert_to_tensor(
          [
              [1, 2, 3],
              [4, 5, 0],
              [6, 0, 0],
          ],
          dtype=tf.int32,
      )
      input_vectors = tf.convert_to_tensor(
          [
              [[0, 1, 2], [1, 2, 3], [2, 3, 4]],
              [[3, 4, 5], [4, 5, 6], [0, 0, 0]],
              [[5, 6, 7], [0, 0, 0], [0, 0, 0]],
          ],
          dtype=tf.float32,
      )
      targets = tf.convert_to_tensor(
          [
              [10, 0, 0],
              [20, 30, 40],
              [50, 60, 0],
          ],
          dtype=tf.int32,
      )
      (
          packed,
          segmentation,
          position,
      ) = pack_sequences_ops.pack_sequences_k(
          [input_tokens, input_vectors, targets], [5, 3, 5]
      )
      (
          input_tokens_packed,
          input_vectors_packed,
          targets_packed,
      ) = packed
      (
          input_tokens_segmentation,
          input_vectors_segmentation,
          targets_segmentation,
      ) = segmentation
      (
          input_tokens_position,
          input_vectors_position,
          targets_position,
      ) = position
      self.assertAllEqual(
          input_tokens_packed,
          [
              [1, 2, 3, 0, 0],
              [4, 5, 6, 0, 0],
          ],
      )
      self.assertAllEqual(
          input_vectors_packed,
          [
              [[0, 1, 2], [1, 2, 3], [2, 3, 4]],
              [[3, 4, 5], [4, 5, 6], [5, 6, 7]],
          ],
      )
      self.assertAllEqual(
          input_tokens_segmentation,
          [
              [1, 1, 1, 0, 0],
              [1, 1, 2, 0, 0],
          ],
      )
      self.assertAllEqual(
          input_vectors_segmentation,
          [
              [1, 1, 1],
              [1, 1, 2],
          ],
      )
      self.assertAllEqual(
          input_tokens_position,
          [
              [0, 1, 2, 0, 0],
              [0, 1, 0, 0, 0],
          ],
      )
      self.assertAllEqual(
          input_vectors_position,
          [
              [0, 1, 2],
              [0, 1, 0],
          ],
      )
      self.assertAllEqual(
          targets_packed,
          [
              [10, 0, 0, 0, 0],
              [20, 30, 40, 50, 60],
          ],
      )
      self.assertAllEqual(
          targets_segmentation,
          [
              [1, 0, 0, 0, 0],
              [1, 1, 1, 2, 2],
          ],
      )
      self.assertAllEqual(
          targets_position,
          [
              [0, 0, 0, 0, 0],
              [0, 1, 2, 0, 1],
          ],
      )
  
imports: from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  import numpy
  from tensor2tensor.data_generators.ops import pack_sequences_ops
  import tensorflow.compat.v1
label_desc: Use image node_id 10 for calling the PackSequencesOpsTest obj's underlying member method code with example usage: obj.test_pack_sequences_k_multi_input() without return types