()
return tl.Fn('InsertAxes12',
               lambda x: jnp.reshape(x, (x.shape[0], 1, 1, x.shape[1])))
Returns a layer that inserts two internal size-1 axes into an array.
Returns a layer that inserts two internal size-1 axes into an array.
Returns a layer that inserts two internal size - 1 axes into an array .
def _InsertAxes12():
  """Returns a layer that inserts two internal size-1 axes into an array."""
  return tl.Fn('InsertAxes12',
               lambda x: jnp.reshape(x, (x.shape[0], 1, 1, x.shape[1])))
def _InsertAxes12 ( ) : return tl . Fn ( 'InsertAxes12' , lambda x : jnp . reshape ( x , ( x . shape [ 0 ] , 1 , 1 , x . shape [ 1 ] ) ) )