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/api_docs/python/tf/experimental/dtensor/unpack