|
| 1 | +using System.Collections.Generic; |
| 2 | + |
| 3 | +namespace Tensorflow.Keras.ArgsDefinition.Rnn |
| 4 | +{ |
| 5 | + public class RNNArgs : LayerArgs |
| 6 | + { |
| 7 | + public interface IRnnArgCell : ILayer |
| 8 | + { |
| 9 | + object state_size { get; } |
| 10 | + } |
| 11 | + |
| 12 | + public IRnnArgCell Cell { get; set; } = null; |
| 13 | + public bool ReturnSequences { get; set; } = false; |
| 14 | + public bool ReturnState { get; set; } = false; |
| 15 | + public bool GoBackwards { get; set; } = false; |
| 16 | + public bool Stateful { get; set; } = false; |
| 17 | + public bool Unroll { get; set; } = false; |
| 18 | + public bool TimeMajor { get; set; } = false; |
| 19 | + public Dictionary<string, object> Kwargs { get; set; } = null; |
| 20 | + |
| 21 | + public int Units { get; set; } |
| 22 | + public Activation Activation { get; set; } |
| 23 | + public Activation RecurrentActivation { get; set; } |
| 24 | + public bool UseBias { get; set; } = true; |
| 25 | + public IInitializer KernelInitializer { get; set; } |
| 26 | + public IInitializer RecurrentInitializer { get; set; } |
| 27 | + public IInitializer BiasInitializer { get; set; } |
| 28 | + |
| 29 | + // kernel_regularizer=None, |
| 30 | + // recurrent_regularizer=None, |
| 31 | + // bias_regularizer=None, |
| 32 | + // activity_regularizer=None, |
| 33 | + // kernel_constraint=None, |
| 34 | + // recurrent_constraint=None, |
| 35 | + // bias_constraint=None, |
| 36 | + // dropout=0., |
| 37 | + // recurrent_dropout=0., |
| 38 | + // return_sequences=False, |
| 39 | + // return_state=False, |
| 40 | + // go_backwards=False, |
| 41 | + // stateful=False, |
| 42 | + // unroll=False, |
| 43 | + // **kwargs): |
| 44 | + } |
| 45 | +} |
0 commit comments