code: def split_tensor_into_1d_equal_chunks(tensor):
      """Break a tensor into equal 1D chunks."""
      data = tensor.view(-1)
      partition_size = (
          torch.numel(data)
          // parallel_state.get_tensor_model_parallel_world_size()
      )
      start_index = (
          partition_size
          * parallel_state.get_tensor_model_parallel_rank()
      )
      end_index = start_index + partition_size
      return data[start_index:end_index]
  
imports: import torch
  from apex.transformer import parallel_state
label_desc: Use image node_id 3 for calling a global function with example usage: split_tensor_into_1d_equal_chunks(tensor) and returns: data