Browse Source

trainable_variables #359

tags/v0.12
Oceania2018 6 years ago
parent
commit
210bc8916e
1 changed files with 8 additions and 0 deletions
  1. +8
    -0
      src/TensorFlowNET.Core/APIs/tf.variable.cs

+ 8
- 0
src/TensorFlowNET.Core/APIs/tf.variable.cs View File

@@ -32,6 +32,14 @@ namespace Tensorflow
return variables.variables_initializer(g.ToArray());
}

/// <summary>
/// Returns all variables created with `trainable=True`.
/// </summary>
/// <param name="scope"></param>
/// <returns></returns>
public VariableV1[] trainable_variables(string scope = null)
=> (variables.trainable_variables() as List<VariableV1>).ToArray();

public RefVariable get_variable(string name,
TensorShape shape = null,
TF_DataType dtype = TF_DataType.DtInvalid,


Loading…
Cancel
Save