tf.raw_ops.TPUPartitionedOutputV2
Операция, демультиплексирующая тензор, который должен быть разбит на шарды XLA, в список секционированных
tf.raw_ops.TPUPartitionedOutputV2(
inputs, num_splits, partition_dims, name=None
)
результатов за пределами вычисления XLA. Поддерживается секционирование ND.
| Аргументы | |
|---|---|
inputs | Tensor. Тензор, представляющий полную форму секционированных тензоров. |
num_splits | int, который >= 1. |
partition_dims | Список ints. Список целых чисел, описывающих, как секционируется каждое измерение. Пустой список означает, что входные данные реплицированы. |
name | Имя операции (необязательно). |
| Возвращает | |
|---|---|
Список объектов num_splits 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/TPUPartitionedOutputV2
