prompt: You will be provided with a partial code base and an issue statement explaining a problem to resolve.
  <issue>
  fft2
  
  
  </issue>
  <code>
  [start of ivy/functional/frontends/jax/numpy/fft.py]
  1 # local
  2 import ivy
  3 from ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back
  4 from ivy.func_wrapper import with_unsupported_dtypes
  5 
  6 
  7 @to_ivy_arrays_and_back
  8 def fft(a, n=None, axis=-1, norm=None):
  9     if norm is None:
  10         norm = "backward"
  11     return ivy.fft(a, axis, norm=norm, n=n)
  12 
  13 
  14 @to_ivy_arrays_and_back
  15 @with_unsupported_dtypes({"2.4.2 and below": ("float16", "bfloat16")}, "paddle")
  16 def fftshift(x, axes=None, name=None):
  17     shape = x.shape
  18 
  19     if axes is None:
  20         axes = tuple(range(x.ndim))
  21         shifts = [(dim // 2) for dim in shape]
  22     elif isinstance(axes, int):
  23         shifts = shape[axes] // 2
  24     else:
  25         shifts = [shape[ax] // 2 for ax in axes]
  26 
  27     roll = ivy.roll(x, shifts, axis=axes)
  28 
  29     return roll
  30
  [end of ivy/functional/frontends/jax/numpy/fft.py]
  </code>
  
  I need you to solve this issue by generating a single patch file that I can apply directly to this repository using git apply. Please respond with a single patch file in the following format.
  <patch>
  diff --git a/file.py b/file.py
  --- a/file.py
  +++ b/file.py
  @@ -1,27 +1,35 @@
   def euclidean(a, b):
  -    while b:
  -        a, b = b, a % b
  -    return a
  +    if b == 0:
  +        return a
  +    return euclidean(b, a % b)
   
   
   def bresenham(x0, y0, x1, y1):
       points = []
       dx = abs(x1 - x0)
       dy = abs(y1 - y0)
  -    sx = 1 if x0 < x1 else -1
  -    sy = 1 if y0 < y1 else -1
  -    err = dx - dy
  +    x, y = x0, y0
  +    sx = -1 if x0 > x1 else 1
  +    sy = -1 if y0 > y1 else 1
   
  -    while True:
  -        points.append((x0, y0))
  -        if x0 == x1 and y0 == y1:
  -            break
  -        e2 = 2 * err
  -        if e2 > -dy:
  -            err -= dy
  -            x0 += sx
  -        if e2 < dx:
  -            err += dx
  -            y0 += sy
  +    if dx > dy:
  +        err = dx / 2.0
  +        while x != x1:
  +            points.append((x, y))
  +            err -= dy
  +            if err < 0:
  +                y += sy
  +                err += dx
  +            x += sx
  +    else:
  +        err = dy / 2.0
  +        while y != y1:
  +            points.append((x, y))
  +            err -= dx
  +            if err < 0:
  +                x += sx
  +                err += dy
  +            y += sy
  +
  +    points.append((x, y))
       return points
  
  </patch>
golden_diff: diff --git a/ivy/functional/frontends/jax/numpy/fft.py b/ivy/functional/frontends/jax/numpy/fft.py
  --- a/ivy/functional/frontends/jax/numpy/fft.py
  +++ b/ivy/functional/frontends/jax/numpy/fft.py
  @@ -11,6 +11,13 @@
       return ivy.fft(a, axis, norm=norm, n=n)
   
   
  +@to_ivy_arrays_and_back
  +def fft2(a, s=None, axes=(-2, -1), norm=None):
  +    if norm is None:
  +        norm = "backward"
  +    return ivy.array(ivy.fft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))
  +
  +
   @to_ivy_arrays_and_back
   @with_unsupported_dtypes({"2.4.2 and below": ("float16", "bfloat16")}, "paddle")
   def fftshift(x, axes=None, name=None):
  
verification_info: {"golden_diff": "diff --git a/ivy/functional/frontends/jax/numpy/fft.py b/ivy/functional/frontends/jax/numpy/fft.py\n--- a/ivy/functional/frontends/jax/numpy/fft.py\n+++ b/ivy/functional/frontends/jax/numpy/fft.py\n@@ -11,6 +11,13 @@\n     return ivy.fft(a, axis, norm=norm, n=n)\n \n \n+@to_ivy_arrays_and_back\n+def fft2(a, s=None, axes=(-2, -1), norm=None):\n+    if norm is None:\n+        norm = \"backward\"\n+    return ivy.array(ivy.fft2(a, s=s, dim=axes, norm=norm), dtype=ivy.dtype(a))\n+\n+\n @to_ivy_arrays_and_back\n @with_unsupported_dtypes({\"2.4.2 and below\": (\"float16\", \"bfloat16\")}, \"paddle\")\n def fftshift(x, axes=None, name=None):\n", "issue": "fft2\n\n", "before_files": [{"content": "# local\nimport ivy\nfrom ivy.functional.frontends.jax.func_wrapper import to_ivy_arrays_and_back\nfrom ivy.func_wrapper import with_unsupported_dtypes\n\n\n@to_ivy_arrays_and_back\ndef fft(a, n=None, axis=-1, norm=None):\n    if norm is None:\n        norm = \"backward\"\n    return ivy.fft(a, axis, norm=norm, n=n)\n\n\n@to_ivy_arrays_and_back\n@with_unsupported_dtypes({\"2.4.2 and below\": (\"float16\", \"bfloat16\")}, \"paddle\")\ndef fftshift(x, axes=None, name=None):\n    shape = x.shape\n\n    if axes is None:\n        axes = tuple(range(x.ndim))\n        shifts = [(dim // 2) for dim in shape]\n    elif isinstance(axes, int):\n        shifts = shape[axes] // 2\n    else:\n        shifts = [shape[ax] // 2 for ax in axes]\n\n    roll = ivy.roll(x, shifts, axis=axes)\n\n    return roll\n", "path": "ivy/functional/frontends/jax/numpy/fft.py"}]}