Spec-Zone.ru › TensorFlow 2.9

tf.experimental.dtensor.unpack

Распаковывает DTensor в tf.Tensor компоненты.

tf.experimental.dtensor.unpack(
    tensor: Any
) -> Sequence[Any]

Упаковывание и распаковывание — это обратные операции:

* unpack(pack(tensors)) == tensors
* pack(unpack(dtensor)) == dtensor
  1. Для любого DTensor на сетке, unpack возвращает исходные компоненты, размещённые на каждом базовом устройстве.
  2. Упаковывание этих исходных компонентов в том же порядке с использованием 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

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API