def METHOD_NAME(self, prev_rank: int = None) -> Any:
    """Copy the forward output from the previous stage in pipeline as the input tensor of this stage.
    Args:
        prev_rank (int, optional): The rank of the source of the tensor.
    Returns:
        Any: The input tensor or input tensor list.
    """
    if prev_rank is None:
        prev_rank = self.stage_manager.get_prev_rank()
    cur_rank = self.stage_manager.get_rank()
    input_tensor = _recv_object(prev_rank, cur_rank, self.stage_manager.get_p2p_process_group(prev_rank, cur_rank))
    return input_tensor