class_docstr: Split the input and keep only the corresponding chuck to the rank.
code: def symbolic(graph, input_):
      return _split_along_last_dim(input_)
  
imports: import torch
  from apex.transformer.parallel_state import get_tensor_model_parallel_group
  from apex.transformer.parallel_state import get_tensor_model_parallel_world_size
  from apex.transformer.parallel_state import get_tensor_model_parallel_rank
  from apex.transformer.tensor_parallel.utils import split_tensor_along_last_dim
label_desc: Use image node_id 1 for calling the _ScatterToModelParallelRegion obj's underlying member method code with example usage: obj.symbolic(graph, input_) and returns: _split_along_last_dim