Unpacks a DTensor into tf.Tensor
components.
tf.experimental.dtensor.unpack( tensor: Any ) -> Sequence[Any]
Packing and unpacking are inverse operations:
* unpack(pack(tensors)) == tensors * pack(unpack(dtensor)) == dtensor
unpack
returns the raw components placed on each underlying device.pack
returns a DTensor which should be identical to the original DTensor--both the content value and the layout.See the documentation for pack
for more information about how packing and unpacking works.
Args | |
---|---|
tensor | The DTensor to unpack. |
Returns | |
---|---|
The individual component tensors of the DTensor. This will include only the client-local components, i.e. the components placed on the local devices. |
Raises | |
---|---|
RuntimeError | When unpack is not called eagerly. |
© 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