parameters: (x, name)
return_statement: return mb.squeeze(x=x_concat, axes=_np.array([0]), name=name)
docstring: Used to transform lstm state values (hn, cn)
          from pytorch to CoreML format.
docstring_summary: Used to transform lstm state values (hn, cn)
          from pytorch to CoreML format.
docstring_tokens: Used to transform lstm state values ( hn cn ) from pytorch to CoreML format .
function: def _pytorch_hidden_to_coreml_milops(x, name):
      """
          Used to transform lstm state values (hn, cn)
          from pytorch to CoreML format.
      """
      split_size = x.shape[0] // 2
      x_split = mb.split(x=x, split_sizes=_np.array([split_size] * 2), axis=0)
      x_concat = mb.concat(values=[x_split[0], x_split[1]], axis=2,)
      # (4.) See docstring to @lstm
      return mb.squeeze(x=x_concat, axes=_np.array([0]), name=name)
function_tokens: def _pytorch_hidden_to_coreml_milops ( x , name ) : split_size = x . shape [ 0 ] // 2 x_split = mb . split ( x = x , split_sizes = _np . array ( [ split_size ] * 2 ) , axis = 0 ) x_concat = mb . concat ( values = [ x_split [ 0 ] , x_split [ 1 ] ] , axis = 2 , ) # (4.) See docstring to @lstm return mb . squeeze ( x = x_concat , axes = _np . array ( [ 0 ] ) , name = name )