tf.dynamic_partition

Разделяет тензор data на список тензоров num_partitions, используя индексы из partitions.

Псевдонимы

Псевдонимы для миграции

См. Руководство по миграции для получения дополнительных сведений.

tf.compat.v1.dynamic_partition

tf.dynamic_partition(
    data: Annotated[Any, TV_DynamicPartition_T],
    partitions: Annotated[Any, _atypes.Int32],
    num_partitions: int,
    name=None
)

Для каждой кортежа индексов js размера partitions.ndim, срез data[js, ...] становится частью outputs[partitions[js]]. Срезы с partitions[js] = i помещаются в outputs[i] в лексикографическом порядке js, и первое измерение outputs[i] равно количеству элементов в partitions, равных i. Подробно,

outputs[i].shape = [sum(partitions == i)] + data.shape[partitions.ndim:]

outputs[i] = pack([data[js, ...] for js if partitions[js] == i])

data.shape должен начинаться с partitions.shape.

Пример:

# Scalar partitions.
partitions = 1
num_partitions = 2
data = [10, 20]
outputs[0] = []  # Empty with shape [0, 2]
outputs[1] = [[10, 20]]

# Vector partitions.
partitions = [0, 0, 1, 1, 0]
num_partitions = 2
data = [10, 20, 30, 40, 50]
outputs[0] = [10, 20, 50]
outputs[1] = [30, 40]

См. dynamic_stitch, чтобы узнать, как объединить разделы обратно.

Исключения
`InvalidArgumentError` в следующих случаях: Если `partitions` не в диапазоне `[0, num_partiions)` Если `partitions.shape` не совпадает с префиксом аргумента `data.shape`.

Аргументы
`data`	A `Tensor`.
`partitions`	A `Tensor` типа `int32`. Любая форма. Индексы в диапазоне `[0, num_partitions)`.
`num_partitions`	A `int`, который является `>= 1`. Количество разделов для вывода.
`name`	Имя операции (необязательно).

Возвращаемое значение
Список объектов `num_partitions` `Tensor` того же типа, что и `data`.

© 2022 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 4.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/api_docs/python/tf/dynamic_partition