override tensorflow %*% S3 method

Author: t-kalinowski

NEWS.md

@@ -11,6 +11,9 @@
 
 - `dotty::.` is now reexported.
 
+- `%*%` now dispatches to `op_matmul()` for tensorflow tensors, which
+  has relaxed shape constraints compared to `tf$matmul()`.
+
 ## Added compatibility with Keras v3.8.0. User-facing changes:
 
 - New symbols:

R/package.R

@@ -56,6 +56,8 @@ keras <- NULL
   requireNamespace("tensorflow", quietly = TRUE)
   maybe_register_S3_methods()
 
+  registerS3method("%*%", "tensorflow.tensor", op_matmul, baseenv())
+
   # if KERAS_PYTHON is defined then forward it to RETICULATE_PYTHON
   keras_python <- get_keras_python()
   if (!is.null(keras_python))