code: def init_tensor(self, shape, init_func, src):
      from fate.arch.protocol.mpc.primitives import (
          ArithmeticSharedTensor,
      )
  
      if self.rank == src:
          return ArithmeticSharedTensor(
              self._ctx, init_func(shape), src=src
          )
  
      else:
          return ArithmeticSharedTensor(
              self._ctx, None, size=shape, src=src
          )
  
imports: import typing
  import warnings
  from typing import overload
  import torch
  from fate.arch.protocol import mpc
  import logging
label_desc: Use image node_id 31 for calling the MPC obj's underlying member method code with example usage: obj.init_tensor(shape, init_func, src) and returns: ArithmeticSharedTensor, ArithmeticSharedTensor