add op_exp2() and op_inner()

Author: t-kalinowski

.tether/man/keras.ops.numpy.txt

@@ -131,6 +131,7 @@ einsum(subscripts, *operands)
 empty(shape, dtype=None)
 equal(x1, x2)
 exp(x)
+exp2(x)
 expand_dims(x, axis)
 expm1(x)
 eye(
@@ -163,6 +164,7 @@ histogram(
 hstack(xs)
 identity(n, dtype=None)
 imag(x)
+inner(x1, x2)
 isclose(
   x1,
   x2,

R/ops-numpy.R

@@ -436,3 +436,52 @@ keras$ops$saturate_cast(x, dtype)
 op_trunc <-
 function (x)
 keras$ops$trunc(x)
+
+
+#' Calculate the base-2 exponential of all elements in the input tensor.
+#'
+#' @returns
+#' Output tensor, element-wise base-2 exponential of `x`.
+#'
+#' @param x
+#' Input tensor.
+#'
+#' @export
+#' @family numpy ops
+#' @family ops
+#' @tether keras.ops.exp2
+op_exp2 <-
+function (x)
+keras$ops$exp2(x)
+
+
+#' Return the inner product of two tensors.
+#'
+#' @description
+#' Ordinary inner product of vectors for 1-D tensors
+#' (without complex conjugation), in higher dimensions
+#' a sum product over the last axes.
+#'
+#' Multidimensional arrays are treated as vectors by flattening
+#' all but their last axes. The resulting dot product is performed
+#' over their last axes.
+#'
+#' @returns
+#' Output tensor. The shape of the output is determined by
+#' broadcasting the shapes of `x1` and `x2` after removing
+#' their last axes.
+#'
+#' @param x1
+#' First input tensor.
+#'
+#' @param x2
+#' Second input tensor. The last dimension of `x1` and `x2`
+#' must match.
+#'
+#' @export
+#' @family numpy ops
+#' @family ops
+#' @tether keras.ops.inner
+op_inner <-
+function (x1, x2)
+keras$ops$inner(x1, x2)

tools/retether.R

@@ -80,27 +80,36 @@ if(FALSE) {
   mk_export("keras.activations.squareplus")$dump |> cat_cb()
   mk_export("keras.activations.tanh_shrink")$dump |> cat_cb()
   mk_export("keras.config.is_flash_attention_enabled")$dump |> cat_cb()
+  mk_export("keras.losses.circle")$dump |> cat_cb()
+  mk_export("keras.initializers.STFT")$dump |> cat_cb()
+  mk_export("keras.metrics.ConcordanceCorrelation")$dump |> cat_cb()
+  mk_export("keras.metrics.PearsonCorrelation")$dump |> cat_cb()
 
 
   mk_export("keras.Model.set_state_tree")$dump |> cat_cb()
   mk_export("keras.layers.Solarization")$dump |> cat_cb()
 
   catched <- character()
   catch <- function(...) catched <<- c(catched, "\n\n", ...)
-  mk_export("keras.ops.bitwise_and")$dump               |> catch()
-  mk_export("keras.ops.bitwise_invert")$dump            |> catch()
-  mk_export("keras.ops.bitwise_left_shift")$dump        |> catch()
-  mk_export("keras.ops.bitwise_not")$dump               |> catch()
-  mk_export("keras.ops.bitwise_or")$dump                |> catch()
-  mk_export("keras.ops.bitwise_right_shift")$dump       |> catch()
-  mk_export("keras.ops.bitwise_xor")$dump               |> catch()
-  mk_export("keras.ops.dot_product_attention")$dump     |> catch()
-  mk_export("keras.ops.histogram")$dump                 |> catch()
-  mk_export("keras.ops.left_shift")$dump                |> catch()
-  mk_export("keras.ops.right_shift")$dump               |> catch()
-  mk_export("keras.ops.logdet")$dump                    |> catch()
-  mk_export("keras.ops.saturate_cast")$dump             |> catch()
-  mk_export("keras.ops.trunc")$dump                     |> catch()
+  mk_export("keras.ops.exp2")$dump               |> catch()
+  mk_export("keras.ops.inner")$dump               |> catch()
+
+  # mk_export("keras.ops.glu")$dump               |> catch()
+  # mk_export("keras.ops.hard_shrink")$dump            |> catch()
+  # mk_export("keras.ops.hard_tanh")$dump        |> catch()
+  # mk_export("keras.ops.soft_shrink")$dump               |> catch()
+  # mk_export("keras.ops.squareplus")$dump                |> catch()
+  # mk_export("keras.ops.tanh_shrink")$dump       |> catch()
+  # mk_export("keras.ops.celu")$dump               |> catch()
+
+  #
+  # mk_export("keras.ops.dot_product_attention")$dump     |> catch()
+  # mk_export("keras.ops.histogram")$dump                 |> catch()
+  # mk_export("keras.ops.left_shift")$dump                |> catch()
+  # mk_export("keras.ops.right_shift")$dump               |> catch()
+  # mk_export("keras.ops.logdet")$dump                    |> catch()
+  # mk_export("keras.ops.saturate_cast")$dump             |> catch()
+  # mk_export("keras.ops.trunc")$dump                     |> catch()
 
 
   catched |>  str_flatten_and_compact_lines(roxygen = TRUE) |> cat_cb()
@@ -120,3 +129,12 @@ view_vignette_adaptation_diff <- function(rmd_file) {
 }
 
 # view_vignette_adaptation_diff("vignettes-src/writing_a_training_loop_from_scratch.Rmd")
+
+add 7 new nn ops
+op_glu()
+op_hard_shrink()
+op_hard_tanh()
+op_soft_shrink()
+op_squareplus()
+op_tanh_shrink()
+op_celu()