tf.растянутый.stack_dynamic_partitions

Складывает динамические разделы тензора или 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'], []]>