tf.experimental.dtensor.unpack
Распаковывает DTensor в компоненты tf.Tensor.
tf.experimental.dtensor.unpack(
tensor: Any
) -> Sequence[Any]
Используется в блокнотах
| Используется в руководстве |
|---|
Упаковка и распаковка — обратные операции:
* unpack(pack(tensors)) == tensors * pack(unpack(dtensor)) == dtensor
- Для любого DTensor на mesh функция
unpackвозвращает исходные компоненты, размещённые на каждом базовом устройстве. - Упаковка этих исходных компонентов в том же порядке с помощью
packвозвращает DTensor, который должен быть идентичен исходному DTensor — как по значению содержимого, так и по компоновке.
Дополнительные сведения о работе упаковки и распаковки см. в документации к pack.
| Аргументы | |
|---|---|
tensor | DTensor для распаковки. |
| Возвращает | |
|---|---|
| Отдельные тензоры-компоненты DTensor. Сюда входят только компоненты, локальные для клиента, то есть компоненты, размещённые на локальных устройствах. |
| Вызывает исключение | |
|---|---|
RuntimeError | Если unpack не вызывается в eager-режиме. |
© 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/experimental/dtensor/unpack