Copies a tf.Tensor onto the DTensor device with the given layout.
tf.experimental.dtensor.copy_to_mesh(
tensor: Any,
layout: tf.experimental.dtensor.Layout,
source_layout: Optional[tf.experimental.dtensor.Layout] = None
) -> tf.Tensor
| Used in the guide | Used in the tutorials |
|---|---|
Copies a regular tf.Tensor onto the DTensor device. Use the mesh attached to layout as target mesh. This method currently only supports replicated layouts, or one-to-one copies for sharded layouts.
| Args | |
|---|---|
tensor | A regular tf.Tensor to be copied as a DTensor. |
layout | Target layout (and mesh) for the result DTensor. |
source_layout | Source layout of the tensor before copy. This argument is deprecated. |
| Returns | |
|---|---|
| A DTensor on the DTensor device with the given layout. |
© 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