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 на mesh функция 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