code: def test_gather_tree_from_array_2d():
      _test_gather_tree_from_array(depth_ndims=2)
  
imports: import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.seq2seq import attention_wrapper
  from tensorflow_addons.seq2seq import beam_search_decoder, gather_tree
label_desc: Use image node_id 6 for calling a global function with example usage: test_gather_tree_from_array_2d() without return types