tf.raw_ops.TPUPartitionedInputV2

Операция, объединяющая список секционированных входных данных. Поддерживает ND-шардирование.

Совместимые псевдонимы для миграции

Подробнее см. в руководстве по миграции.

tf.raw_ops.TPUPartitionedInputV2(
    inputs, partition_dims, is_packed=False, name=None
)

Аргументы
`inputs`	Список как минимум из 1 объекта `Tensor` одного типа. Список секционированных входных данных, которые должны иметь одинаковую форму.
`partition_dims`	Список значений `ints`. Список целых чисел, описывающих, на сколько частей секционировано каждое измерение. Пустой список указывает, что входные данные реплицированы.
`is_packed`	Необязательный параметр `bool`. По умолчанию — `False`. Указывает, является ли входной ресурс упакованным.
`name`	Имя операции (необязательно).

Возвращает
Значение `Tensor`. Имеет тот же тип, что и `inputs`.

© 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/raw_ops/TPUPartitionedInputV2