tf.experimental.dtensor.copy_to_mesh

Копирует tf.Tensor на устройство DTensor с заданной компоновкой.

tf.experimental.dtensor.copy_to_mesh(
    tensor: Any,
    layout: tf.experimental.dtensor.Layout,
    source_layout: Optional[tf.experimental.dtensor.Layout] = None
) -> tf.Tensor

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

Используется в руководстве	Используется в учебниках
Концепции DTensor	Использование DTensor с Keras Распределенное обучение с DTensor

Копирует обычный tf.Tensor на устройство DTensor. Используйте подключенную к layout сетку в качестве целевой сетки. Этот метод в настоящее время поддерживает только реплицированные компоновки или однозначные копии для фрагментированных компоновок.

Аргументы
`tensor`	Обычный tf.Tensor, который необходимо скопировать как DTensor.
`layout`	Целевая компоновка (и сетка) для результирующего DTensor.
`source_layout`	Исходная компоновка тензора перед копированием. Этот аргумент устарел.

Возвращаемые значения
DTensor на устройстве DTensor с заданной компоновкой.

© 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/copy_to_mesh