Method
__init__
(self,
out_dim,
activation=None,
initializer=lambda: tf.uniform_unit_scaling_initializer(
factor=0.36),
**kwargs)
Source from the content-addressed store, hash-verified
| 244 | |
| 245 | class LSTMLayer(Layer): |
| 246 | def __init__(self, |
| 247 | out_dim, |
| 248 | activation=None, |
| 249 | initializer=lambda: tf.uniform_unit_scaling_initializer( |
| 250 | factor=0.36), |
| 251 | **kwargs): |
| 252 | super(LSTMLayer, self).__init__(**kwargs) |
| 253 | self.out_dim = out_dim |
| 254 | self.activation = activation |
| 255 | self.initializer = initializer |
| 256 | |
| 257 | def build(self, input_shape): |
| 258 | self.lstm_cell = tf.nn.rnn_cell.LSTMCell( |
Callers
nothing calls this directly
Tested by
no test coverage detected