tf.experimental.dtensor.unpack

Unpacks a DTensor into tf.Tensor components.

Used in the notebooks

Used in the guide

Packing and unpacking are inverse operations:

* unpack(pack(tensors)) == tensors
* pack(unpack(dtensor)) == dtensor
  1. For any DTensor on the mesh, unpack returns the raw components placed on each underlying device.
  2. 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.

tensor The DTensor to unpack.

The individual component tensors of the DTensor. This will include only the client-local components, i.e. the components placed on the local devices.

RuntimeError When unpack is not called eagerly.