code: def test_tensor_move():
      spawn(run_tensor_move, 1)
  
imports: import torch
  import colossalai
  from colossalai.legacy.zero.gemini.tensor_utils import colo_model_data_tensor_move, colo_model_data_tensor_move_inline
  from colossalai.legacy.zero.sharded_param import ShardedTensor
  from colossalai.testing import rerun_if_address_is_in_use, spawn
label_desc: Use image node_id 2 for calling a global function with example usage: test_tensor_move() without return types