class_docstr: Strategy extension that support both single and multi worker strategy.
code: def __init__(self, container_strategy, mesh):
      super().__init__(container_strategy)
      self._mesh = mesh
      self._num_clients = d_config.num_clients()
      self._client_id = d_config.client_id()
  
imports: import functools
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import config
  from tensorflow.dtensor.python import d_variable
  from tensorflow.dtensor.python import input_util
  from tensorflow.dtensor.python import layout
  from tensorflow.python.data.experimental.ops import distribute
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import distribute_utils
  from tensorflow.python.distribute.experimental import dtensor_util
  from tensorflow.python.framework import ops
  from tensorflow.python.ops import array_ops
  from tensorflow.python.util import nest
label_desc: Use image node_id 1 to create a new DTensorStrategyExtended object from inherited base classes: distribute_lib with example: obj = DTensorStrategyExtended(container_strategy, mesh)