tf.experimental.dtensor.unpack
Распаковывает DTensor в tf.Tensor компоненты.
tf.experimental.dtensor.unpack(
tensor: Any
) -> Sequence[Any]
Упаковывание и распаковывание — это обратные операции:
* unpack(pack(tensors)) == tensors * pack(unpack(dtensor)) == dtensor
- Для любого DTensor на сетке,
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/versions/r2.9/api_docs/python/tf/experimental/dtensor/unpack