From c4c6dbe2e50348b1fc2de8376357d6ccfbf5beb1 Mon Sep 17 00:00:00 2001 From: Oceania2018 Date: Thu, 17 Oct 2019 09:08:37 -0500 Subject: [PATCH] all tensor learning_rate for tf.train.GradientDescentOptimizer --- src/TensorFlowNET.Core/APIs/tf.train.cs | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/TensorFlowNET.Core/APIs/tf.train.cs b/src/TensorFlowNET.Core/APIs/tf.train.cs index 3a790327..03b0a0e2 100644 --- a/src/TensorFlowNET.Core/APIs/tf.train.cs +++ b/src/TensorFlowNET.Core/APIs/tf.train.cs @@ -34,6 +34,9 @@ namespace Tensorflow public Optimizer GradientDescentOptimizer(float learning_rate) => new GradientDescentOptimizer(learning_rate); + public Optimizer GradientDescentOptimizer(Tensor learning_rate) + => new GradientDescentOptimizer(learning_rate); + public Optimizer AdamOptimizer(float learning_rate, string name = "Adam") => new AdamOptimizer(learning_rate, name: name);