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_):  # type: ignore
      return _split(input_)
  
imports: from typing import Any
  import torch
  from .initialize import get_model_parallel_group
  from .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.forward(ctx, input_) and returns: _split