(self, x)
return fastmath.nested_map(lambda y: y[0], x)
Returns a single-device version of ``x``.
Returns a single-device version of ``x``.
Returns a single - device version of x .
def _unreplicate(self, x):
    """Returns a single-device version of ``x``."""
    if self._n_devices < 2:
      return x
    return fastmath.nested_map(lambda y: y[0], x)
def _unreplicate ( self , x ) : if self . _n_devices < 2 : return x return fastmath . nested_map ( lambda y : y [ 0 ] , x )