add avg_pool_grad function

dogvane · dogvane · commit b968fd79ab15 · 2023-07-10T00:41:23.000+08:00
diff --git a/src/TensorFlowNET.Core/Gradients/nn_grad.cs b/src/TensorFlowNET.Core/Gradients/nn_grad.cs
@@ -365,6 +365,23 @@ public static Tensor[] _MaxPoolGrad(Operation op, Tensor[] grads)
             };
         }
 
+        [RegisterGradient("AvgPool")]
+        public static Tensor[] _AvgPoolGrad(Operation op, Tensor[] grads)
+        {
+            Tensor grad = grads[0];
+            
+            return new Tensor[]
+            {
+                gen_nn_ops.avg_pool_grad(
+                  array_ops.shape(op.inputs[0]),
+                  grad,
+                  op.get_attr_list<int>("ksize"),
+                  op.get_attr_list<int>("strides"),
+                  op.get_attr("padding").ToString(),
+                  op.get_attr("data_format").ToString())
+            };
+        }
+
         /// <summary>
         /// Return the gradients for TopK.
         /// </summary>
