Browse Source

fix: add the GRU, LSTM, SimpleRNN's OptionalArgs

tags/v0.150.0-BERT-Model
Wanglongzhi2001 1 year ago
parent
commit
3fcc4d8d15
3 changed files with 23 additions and 3 deletions
  1. +1
    -3
      src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/GRUOptionalArgs.cs
  2. +11
    -0
      src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/LSTMOptionalArgs.cs
  3. +11
    -0
      src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/SimpleRNNOptionalArgs.cs

+ 1
- 3
src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/GRUOptionalArgs.cs View File

@@ -4,10 +4,8 @@ using System.Text;

namespace Tensorflow.Keras.ArgsDefinition
{
public class GRUOptionalArgs
public class GRUOptionalArgs : RnnOptionalArgs
{
public string Identifier => "GRU";

public Tensor Mask { get; set; } = null;
}
}

+ 11
- 0
src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/LSTMOptionalArgs.cs View File

@@ -0,0 +1,11 @@
using System;
using System.Collections.Generic;
using System.Text;

namespace Tensorflow.Keras.ArgsDefinition.Rnn
{
public class LSTMOptionalArgs : RnnOptionalArgs
{
public string Identifier => "LSTM";
}
}

+ 11
- 0
src/TensorFlowNET.Core/Keras/ArgsDefinition/Rnn/SimpleRNNOptionalArgs.cs View File

@@ -0,0 +1,11 @@
using System;
using System.Collections.Generic;
using System.Text;

namespace Tensorflow.Keras.ArgsDefinition.Rnn
{
public class SimpleRNNOptionalArgs : RnnOptionalArgs
{
public string Identifier => "SimpleRNN";
}
}

Loading…
Cancel
Save