diff --git a/src/TensorFlowNET.Core/APIs/tf.variable.cs b/src/TensorFlowNET.Core/APIs/tf.variable.cs index e855535a..8e905b1d 100644 --- a/src/TensorFlowNET.Core/APIs/tf.variable.cs +++ b/src/TensorFlowNET.Core/APIs/tf.variable.cs @@ -32,6 +32,14 @@ namespace Tensorflow return variables.variables_initializer(g.ToArray()); } + /// + /// Returns all variables created with `trainable=True`. + /// + /// + /// + public VariableV1[] trainable_variables(string scope = null) + => (variables.trainable_variables() as List).ToArray(); + public RefVariable get_variable(string name, TensorShape shape = null, TF_DataType dtype = TF_DataType.DtInvalid,