code: def _band_part(
      input, num_lower, num_upper, name=None
  ):  # pylint: disable=redefined-builtin
      del name
      result = ops.convert_to_tensor(input)
      if num_lower > -1:
          result = np.triu(result, -num_lower)
      if num_upper > -1:
          result = np.tril(result, num_upper)
      return result
  
imports: import collections
  import functools
  import numpy
  from tensorflow_probability.python.internal.backend.numpy import _utils
  from tensorflow_probability.python.internal.backend.numpy import ops
label_desc: Use image node_id 1 for calling a global function with example usage: _band_part(input, num_lower, num_upper, name) and returns: result