()
return tl.Fn('RemoveAxes12', lambda x: jnp.squeeze(x, (1, 2)))
Returns a layer that removes two internal size-1 axes from an array.
Returns a layer that removes two internal size-1 axes from an array.
Returns a layer that removes two internal size - 1 axes from an array .
def _RemoveAxes12():
  """Returns a layer that removes two internal size-1 axes from an array."""
  return tl.Fn('RemoveAxes12', lambda x: jnp.squeeze(x, (1, 2)))
def _RemoveAxes12 ( ) : return tl . Fn ( 'RemoveAxes12' , lambda x : jnp . squeeze ( x , ( 1 , 2 ) ) )