code: def _transpose_batch_time(x):
      return np.transpose(x, [1, 0, 2]).astype(np.int32)
  
file_path: addons/tensorflow_addons/seq2seq/tests/beam_search_ops_test.py
imports: import itertools
  import numpy
  import pytest
  import tensorflow
  from tensorflow_addons.seq2seq import gather_tree
label_desc: Use image node_id 1 for calling a global function with example usage: _transpose_batch_time(x) and returns: np