tf.ragged.stack_dynamic_partitions

Объединяет динамические разделы Tensor или RaggedTensor.

tf.ragged.stack_dynamic_partitions(
    data, partitions, num_partitions, name=None
)

Возвращает RaggedTensor output с num_partitions строками, где строка output[i] сформирована путем объединения всех срезов data[j1...jN], для которых выполняется условие partitions[j1...jN] = i. Срезы data объединяются в порядке следования строк.

Если num_partitions — это int (а не Tensor), то это эквивалентно tf.ragged.stack(tf.dynamic_partition(data, partitions, num_partitions)).

Пример:

data           = ['a', 'b', 'c', 'd', 'e']
partitions     = [  3,   0,   2,   2,   3]
num_partitions = 5
tf.ragged.stack_dynamic_partitions(data, partitions, num_partitions)
<tf.RaggedTensor [[b'b'], [], [b'c', b'd'], [b'a', b'e'], []]>