code: def _get_batch_non_diagonal_gpu(array):
      batch_size, m, n = array.shape
      assert m == n
      return cuda.cupy.ElementwiseKernel(
          "raw T mat, int32 idx",
          "T val",
          "val = mat[idx]",
          "lower_triangular_matrix_get_non_diag",
      )(array, _non_diagonal_idx_array(batch_size, n))
  
imports: from functools import lru_cache
  from chainer import cuda
  from chainer import function
  from chainer.utils import type_check
  import numpy
label_desc: Use image node_id 12 for calling a global function with example usage: _get_batch_non_diagonal_gpu(array) and returns: cuda