Browse Source

Refine some code after merge.

pull/976/head
AsakusaRinne 2 years ago
parent
commit
59c17705c3
2 changed files with 5 additions and 5 deletions
  1. +3
    -3
      src/TensorFlowNET.Core/Operations/Initializers/Orthogonal.cs
  2. +2
    -2
      test/TensorFlowNET.Keras.UnitTest/InitializerTest.cs

+ 3
- 3
src/TensorFlowNET.Core/Operations/Initializers/Orthogonal.cs View File

@@ -32,10 +32,10 @@ public class Orthogonal : IInitializer
_seed = seed; _seed = seed;
} }


private readonly Dictionary<string, object> _config;
private readonly Dictionary<string, object> _config;


public string ClassName => "Orthogonal";
public IDictionary<string, object> Config => throw new NotImplementedException();
public string ClassName => "Orthogonal";
public IDictionary<string, object> Config => throw new NotImplementedException();
public Tensor Apply(InitializerArgs args) public Tensor Apply(InitializerArgs args)
{ {
return _generate_init_val(args.Shape, args.DType == TF_DataType.DtInvalid ? TF_DataType.TF_FLOAT : args.DType); return _generate_init_val(args.Shape, args.DType == TF_DataType.DtInvalid ? TF_DataType.TF_FLOAT : args.DType);


+ 2
- 2
test/TensorFlowNET.Keras.UnitTest/InitializerTest.cs View File

@@ -6,7 +6,7 @@ using System.Text;
using TensorFlowNET.Keras.UnitTest; using TensorFlowNET.Keras.UnitTest;
using static Tensorflow.Binding; using static Tensorflow.Binding;


namespace Tensorflow.Keras.UnitTest;
namespace TensorFlowNET.Keras.UnitTest;


[TestClass] [TestClass]
public class InitializerTest : EagerModeTestBase public class InitializerTest : EagerModeTestBase
@@ -15,6 +15,6 @@ public class InitializerTest : EagerModeTestBase
public void Orthogonal() public void Orthogonal()
{ {
var initializer = tf.keras.initializers.Orthogonal(); var initializer = tf.keras.initializers.Orthogonal();
var values = initializer.Apply(new InitializerArgs((2, 2)));
var values = initializer.Apply(new Tensorflow.InitializerArgs((2, 2)));
} }
} }

Loading…
Cancel
Save