code: def test_distribute_values_from_function(self):
      array_value = np.array(
          [1.0, 2.0, 3.0, 4.0, 5.0, 6.0, 7.0, 8.0, 9.0]
      )
  
      def value_fn(ctx):
          return array_value[ctx.replica_id_in_sync_group]
  
      strategy = mwms.MultiWorkerMirroredStrategy()
      distributed_values = (
          strategy.experimental_distribute_values_from_function(
              value_fn
          )
      )
  
      self.assertEqual(
          d_api.fetch_layout(distributed_values),
          layout.Layout.batch_sharded(
              strategy.mesh, batch_dim="batch", rank=1
          ),
      )
      unpacked_value = d_api.unpack(distributed_values)
      self.assertLen(unpacked_value, self.num_local_devices)
      start = 1.0 + self.num_local_devices * self.client_id
      for i in range(self.num_local_devices):
          self.assertEqual(unpacked_value[i], start + i)
  
imports: import json
  import os
  from absl import flags
  from absl.testing import parameterized
  import numpy
  from tensorflow.dtensor.python import api
  from tensorflow.dtensor.python import d_variable
  from tensorflow.dtensor.python import layout
  from tensorflow.dtensor.python.tests import multi_client_test_util
  from tensorflow.dtensor.python.tests import test_backend_util
  from tensorflow.dtensor.python.tests import test_util
  from tensorflow.python.data.ops import dataset_ops
  from tensorflow.python.distribute import distribute_lib
  from tensorflow.python.distribute import reduce_util
  from tensorflow.python.distribute.cluster_resolver import tfconfig_cluster_resolver
  from tensorflow.python.distribute.experimental import dtensor_util
  from tensorflow.python.distribute.experimental import multi_worker_mirrored_strategy
  from tensorflow.python.eager import def_function
  from tensorflow.python.framework import config
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import tensor_spec
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import math_ops
  from tensorflow.python.ops import stateless_random_ops
  from tensorflow.python.ops import variables
  from tensorflow.python.platform import test
label_desc: Use image node_id 6 for calling the StrategyDatasetTest obj's underlying member method code with example usage: obj.test_distribute_values_from_function() and returns: array_value