code: def __init__(self, tensor: torch.Tensor, waitfunc):
      self.tensor = tensor
      self.waitfunc = waitfunc
  
imports: from functools import reduce
  import operator
  from typing import Union, Optional, Tuple
  import torch
  from apex.transformer import parallel_state
  from apex.transformer.log_util import get_transformer_logger
  from apex.transformer.utils import split_tensor_into_1d_equal_chunks
  from apex.transformer.utils import gather_split_1d_tensor
  from apex.transformer.pipeline_parallel.utils import Shape
  from apex.transformer.pipeline_parallel._timers import _Timers
label_desc: Use image node_id 1 to create a new FutureTensor object  with example: obj = FutureTensor(tensor, waitfunc)