code: def _extract_gates(x):
      r = x.reshape((x.shape[0], x.shape[1] // 4, 4) + x.shape[2:])
      return (r[:, :, i] for i in six.moves.range(4))
  
imports: import numpy
  import six
  from chainer import backend
  from chainer.backends import cuda
  from chainer.backends import intel64
  from chainer import function
  from chainer import function_node
  from chainer.utils import type_check
  import chainerx
label_desc: Use image node_id 1 for calling a global function with example usage: _extract_gates(x) and returns: unknown