We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 8574881 commit b968fd7Copy full SHA for b968fd7
src/TensorFlowNET.Core/Gradients/nn_grad.cs
@@ -365,6 +365,23 @@ public static Tensor[] _MaxPoolGrad(Operation op, Tensor[] grads)
365
};
366
}
367
368
+ [RegisterGradient("AvgPool")]
369
+ public static Tensor[] _AvgPoolGrad(Operation op, Tensor[] grads)
370
+ {
371
+ Tensor grad = grads[0];
372
+
373
+ return new Tensor[]
374
375
+ gen_nn_ops.avg_pool_grad(
376
+ array_ops.shape(op.inputs[0]),
377
+ grad,
378
+ op.get_attr_list<int>("ksize"),
379
+ op.get_attr_list<int>("strides"),
380
+ op.get_attr("padding").ToString(),
381
+ op.get_attr("data_format").ToString())
382
+ };
383
+ }
384
385
/// <summary>
386
/// Return the gradients for TopK.
387
/// </summary>
0 commit comments