code: def _np_shift(values, axis, delta):
      values = np.roll(values, delta, axis)
      sl = [slice(None)] * values.ndim
      if delta > 0:
          sl[axis] = slice(None, delta)
      else:
          sl[axis] = slice(delta, None)
      values[tuple(sl)] = 0
      return values
  
file_path: tf-quant-finance/tf_quant_finance/math/pde/steppers/douglas_adi_scheme_test.py
imports: import numpy
  import tensorflow.compat.v2
  import tf_quant_finance
  from tensorflow.python.framework import test_util
label_desc: Use image node_id 1 for calling a global function with example usage: _np_shift(values, axis, delta) and returns: values