add op_ifft2

Author: t-kalinowski

.tether/man/keras.ops.txt

@@ -289,6 +289,7 @@ histogram(
 )
 hstack(xs)
 identity(n, dtype=None)
+ifft2(x)
 imag(x)
 image: Module(keras.api.ops.image)
 in_top_k(

R/ops.R

@@ -899,6 +899,35 @@ function (x)
 keras$ops$fft2(x)
 
 
+#' Computes the 2D Inverse Fast Fourier Transform along the last two axes of
+#'
+#' @description
+#' input.
+#'
+#' # Examples
+#' ```{r}
+#' x <- list(op_array(rbind(c(1, 2), c(2, 1))),
+#'           op_array(rbind(c(0, 1), c(1, 0))))
+#' op_ifft2(x)
+#' ```
+#'
+#' @returns
+#' A tuple containing two tensors - the real and imaginary parts of the
+#' output.
+#'
+#' @param x
+#' Tuple of the real and imaginary parts of the input tensor. Both
+#' tensors in the tuple should be of floating type.
+#'
+#' @export
+#' @family math ops
+#' @family ops
+#' @tether keras.ops.ifft2
+op_ifft2 <-
+function (x)
+keras$ops$ifft2(tuple(x))
+
+
 #' Checks if the targets are in the top-k predictions.
 #'
 #' @description

tools/retether.R

@@ -88,6 +88,7 @@ if(FALSE) {
 
   mk_export("keras.Model.set_state_tree")$dump |> cat_cb()
   mk_export("keras.layers.Solarization")$dump |> cat_cb()
+  mk_export("keras.ops.ifft2")$dump |> cat_cb()
 
   catched <- character()
   catch <- function(...) catched <<- c(catched, "\n\n", ...)