code: def as_in_context(self, context):
          
          if self.context == context:
              return self
          return self.copyto(context)
docs: Returns an array on the target device with the same value as this array.
  
          If the target context is the same as ``self.context``, then ``self`` is
          returned.  Otherwise, a copy is made.
  
          Parameters
          ----------
          context : Context
              The target context.
  
          Returns
          -------
          NDArray, CSRNDArray or RowSparseNDArray
              The target array.
  
  
          Examples
          --------
          >>> x = mx.nd.ones((2,3))
          >>> y = x.as_in_context(mx.cpu())
          >>> y is x
          True
          >>> z = x.as_in_context(mx.gpu(0))
          >>> z is x
          False