Browse Source

tf.variance_scaling_initializer #396

tags/v0.12
Oceania2018 6 years ago
parent
commit
29b5898792
1 changed files with 20 additions and 0 deletions
  1. +20
    -0
      src/TensorFlowNET.Core/APIs/tf.init.cs

+ 20
- 0
src/TensorFlowNET.Core/APIs/tf.init.cs View File

@@ -62,5 +62,25 @@ namespace Tensorflow
stddev: stddev,
seed: seed,
dtype: dtype);

/// <summary>
/// Initializer capable of adapting its scale to the shape of weights tensors.
/// </summary>
/// <param name="scale"></param>
/// <param name="mode"></param>
/// <param name="distribution"></param>
/// <param name="seed"></param>
/// <param name="dtype"></param>
/// <returns></returns>
public IInitializer variance_scaling_initializer(float scale = 1.0f,
string mode = "fan_in",
string distribution = "truncated_normal",
int? seed = null,
TF_DataType dtype = TF_DataType.TF_FLOAT) => new VarianceScaling(
scale: scale,
mode: mode,
distribution: distribution,
seed: seed,
dtype: dtype);
}
}

Loading…
Cancel
Save