SciSharp / TensorFlow.NET

.NET Standard bindings for Google's TensorFlow for developing, training and deploying Machine Learning models in C# and F#.
https://scisharp.github.io/tensorflow-net-docs
Apache License 2.0
3.17k stars 507 forks source link

fix: optimize some APIs #1129

Closed Beacontownfc closed 1 year ago

Beacontownfc commented 1 year ago

tf.nn.moments : There may be a problem where the gradient is null during the training, it seems that removing the 681st line of code from the [TensorFlowNET.Core/Operations/array_ops.cs] can solve this issue.

tf.nn.batch_normalization: Use nn_impl.batch_normalization to reuse code.