Unpacks a DTensor into tf.Tensor
components.
tf.experimental.dtensor.unpack(
tensor: Any
) -> Sequence[Any]
Used in the notebooks
Used in the guide |
---|
Packing and unpacking are inverse operations:
* unpack(pack(tensors)) == tensors
* pack(unpack(dtensor)) == dtensor
- For any DTensor on the mesh,
unpack
returns the raw components placed on each underlying device. - Packing these raw components in the same order using
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.
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. |