Spec-Zone.ru › TensorFlow

tf.experimental.dtensor.unpack

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

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

Используется в блокнотах

Используется в руководстве
  • Концепции DTensor

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

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

Spec-Zone.ru

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