Skip to content

Commit 3fcc4d8

Browse files
fix: add the GRU, LSTM, SimpleRNN's OptionalArgs
1 parent a73694a commit 3fcc4d8

File tree

3 files changed

+23
-3
lines changed

3 files changed

+23
-3
lines changed

src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/GRUOptionalArgs.cs

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,10 +4,8 @@
44

55
namespace Tensorflow.Keras.ArgsDefinition
66
{
7-
public class GRUOptionalArgs
7+
public class GRUOptionalArgs : RnnOptionalArgs
88
{
99
public string Identifier => "GRU";
10-
11-
public Tensor Mask { get; set; } = null;
1210
}
1311
}
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
5+
namespace Tensorflow.Keras.ArgsDefinition.Rnn
6+
{
7+
public class LSTMOptionalArgs : RnnOptionalArgs
8+
{
9+
public string Identifier => "LSTM";
10+
}
11+
}
Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,11 @@
1+
using System;
2+
using System.Collections.Generic;
3+
using System.Text;
4+
5+
namespace Tensorflow.Keras.ArgsDefinition.Rnn
6+
{
7+
public class SimpleRNNOptionalArgs : RnnOptionalArgs
8+
{
9+
public string Identifier => "SimpleRNN";
10+
}
11+
}

0 commit comments

Comments
 (0)