code: def var_init(max_stack_depth, initial_value):
      type_ = inst.TensorType(
          initial_value.dtype, initial_value.shape[1:]
      )
      var = NP_BACKEND.create_variable(
          None,
          inst.VariableAllocation.FULL,
          type_,
          max_stack_depth,
          batch_size=initial_value.shape[0],
      )
      return var.update(
          initial_value, NP_BACKEND.full_mask(initial_value.shape[0])
      )
  
file_path: probability/tensorflow_probability/python/experimental/auto_batching/numpy_backend_test.py
imports: import hypothesis
  from hypothesis import strategies
  from hypothesis.extra import numpy
  import numpy
  from tensorflow_probability.python.experimental.auto_batching import backend_test_lib
  from tensorflow_probability.python.experimental.auto_batching import instructions
  from tensorflow_probability.python.experimental.auto_batching import numpy_backend
  from tensorflow_probability.python.internal import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: var_init(max_stack_depth, initial_value) and returns: var