tf.nest.pack_sequence_as

Возвращает заданную плоскую последовательность, упакованную в заданную структуру.

tf.nest.pack_sequence_as(
    structure, flat_sequence, expand_composites=False
)

Используется в блокнотах

Определение структуры см. в разделе tf.nest.

Если structure — атом, flat_sequence должен быть списком из одного элемента; в этом случае возвращаемое значение — flat_sequence[0].

Если structure является экземпляром dict или содержит его, ключи будут отсортированы, чтобы упаковать плоскую последовательность в детерминированном порядке. Это также относится к экземплярам OrderedDict: порядок элементов в последовательности не учитывается, вместо него используется порядок сортировки ключей. В flatten применяется то же соглашение. Это позволяет корректно упаковывать обратно словари и OrderedDict после их преобразования в плоскую последовательность, а также преобразовать OrderedDict в плоскую последовательность, а затем упаковать обратно с помощью соответствующего обычного словаря, или наоборот. Словари с несортируемыми ключами преобразовать в плоскую последовательность нельзя.

Примеры:

1. Словарь Python:

   structure = { "key3": "", "key1": "", "key2": "" }
   flat_sequence = ["value1", "value2", "value3"]
   tf.nest.pack_sequence_as(structure, flat_sequence)
     {'key3': 'value3', 'key1': 'value1', 'key2': 'value2'}
     

2. Вложенный кортеж Python:

   structure = (('a','b'), ('c','d','e'), 'f')
   flat_sequence = [1.0, 2.0, 3.0, 4.0, 5.0, 6.0]
   tf.nest.pack_sequence_as(structure, flat_sequence)
     ((1.0, 2.0), (3.0, 4.0, 5.0), 6.0)
     

3. Вложенный словарь словарей:

   structure = { "key3": {"c": ('alpha', 'beta'), "a": ('gamma')},
                 "key1": {"e": "val1", "d": "val2"} }
   flat_sequence = ['val2', 'val1', 3.0, 1.0, 2.0]
   tf.nest.pack_sequence_as(structure, flat_sequence)
     {'key3': {'c': (1.0, 2.0), 'a': 3.0}, 'key1': {'e': 'val1', 'd': 'val2'} }
     

4. Массив Numpy (считается скаляром):

   structure = ['a']
   flat_sequence = [np.array([[1, 2], [3, 4]])]
   tf.nest.pack_sequence_as(structure, flat_sequence)
     [array([[1, 2],
            [3, 4]])]
     

5. tf.Tensor (считается скаляром):

   structure = ['a']
   flat_sequence = [tf.constant([[1., 2., 3.], [4., 5., 6.]])]
   tf.nest.pack_sequence_as(structure, flat_sequence)
     [<tf.Tensor: shape=(2, 3), dtype=float32,
      numpy= array([[1., 2., 3.], [4., 5., 6.]], dtype=float32)>]
     

6. tf.RaggedTensor: это составной тензор, представление которого состоит из плоского списка «values» и списка «row_splits», указывающего, как разделить плоский список на строки. Подробнее о tf.RaggedTensor см. на странице https://www.tensorflow.org/api_docs/python/tf/RaggedTensor.

С expand_composites=False мы рассматриваем RaggedTensor как скаляр.

structure = { "foo": tf.ragged.constant([[1, 2], [3]]),
              "bar": tf.constant([[5]]) }
flat_sequence = [ "one", "two" ]
tf.nest.pack_sequence_as(structure, flat_sequence,
expand_composites=False)
  {'foo': 'two', 'bar': 'one'}
  

С expand_composites=True мы предполагаем, что плоский вход содержит тензоры, из которых состоит ragged tensor, то есть тензоры values и row_splits.

structure = { "foo": tf.ragged.constant([[1., 2.], [3.]]),
              "bar": tf.constant([[5.]]) }
tensors = tf.nest.flatten(structure, expand_composites=True)
print(tensors)
  [<tf.Tensor: shape=(1, 1), dtype=float32, numpy=array([[5.]],
   dtype=float32)>,
   <tf.Tensor: shape=(3,), dtype=float32, numpy=array([1., 2., 3.],
   dtype=float32)>,
   <tf.Tensor: shape=(3,), dtype=int64, numpy=array([0, 2, 3])>]
verified_tensors = [tf.debugging.check_numerics(t, 'invalid tensor: ')
                    if t.dtype==tf.float32 else t
                    for t in tensors]
tf.nest.pack_sequence_as(structure, verified_tensors,
                         expand_composites=True)
  {'foo': <tf.RaggedTensor [[1.0, 2.0], [3.0]]>,
   'bar': <tf.Tensor: shape=(1, 1), dtype=float32, numpy=array([[5.]],
   dtype=float32)>}