From ed33b7329ba891f4f350ed9b906c3260daa39b72 Mon Sep 17 00:00:00 2001
From: RayWang <75263275+RayWang-iat@users.noreply.github.com>
Date: Fri, 2 Jun 2023 23:16:38 +0800
Subject: [PATCH] Update Numpy.Math.cs

---
 src/TensorFlowNET.Core/NumPy/Numpy.Math.cs | 11 ++++++++++-
 1 file changed, 10 insertions(+), 1 deletion(-)

diff --git a/src/TensorFlowNET.Core/NumPy/Numpy.Math.cs b/src/TensorFlowNET.Core/NumPy/Numpy.Math.cs
index 0e50cd56..ea85048f 100644
--- a/src/TensorFlowNET.Core/NumPy/Numpy.Math.cs
+++ b/src/TensorFlowNET.Core/NumPy/Numpy.Math.cs
@@ -28,7 +28,16 @@ namespace Tensorflow.NumPy
         public static NDArray multiply(NDArray x1, NDArray x2) => new NDArray(tf.multiply(x1, x2));
 
         [AutoNumPy]
-        public static NDArray maximum(NDArray x1, NDArray x2) => new NDArray(tf.maximum(x1, x2));
+        //public static NDArray maximum(NDArray x1, NDArray x2) => new NDArray(tf.maximum(x1, x2));
+        public static NDArray maximum(NDArray x1, NDArray x2, int? axis = null)
+        {
+            var maxValues = tf.maximum(x1, x2);
+            if (axis.HasValue)
+            {
+                maxValues = tf.reduce_max(maxValues, axis: axis.Value);
+            }
+            return new NDArray(maxValues);
+        }
 
         [AutoNumPy]
         public static NDArray minimum(NDArray x1, NDArray x2) => new NDArray(tf.minimum(x1, x2));