class_docstr: Split the input and keep only the corresponding chuck to the rank.
class_docstr_tok: ["Split","the","input","and","keep","only","the","corresponding","chuck","to","the","rank","."]
code: def forward(ctx, input_):
      return _split_along_first_dim(input_)
  
imports: import torch
  from megatron.core.parallel_state import get_tensor_and_expert_parallel_group, get_tensor_model_parallel_group, get_tensor_model_parallel_rank, get_tensor_model_parallel_world_size
  from .utils import split_tensor_along_last_dim
label_desc: Use image node_id 2 for calling the _ScatterToSequenceParallelRegion obj's underlying member method code with example usage: obj.forward(ctx, input_) and returns: _split_along_first_dim