update docstring in lstm.py
@@ -57,9 +57,9 @@ class StatelessLSTM(LSTMBase):
hidden states.
Args:
- in_size (int): Dimension of input vectors. If ``None``, parameter
- initialization will be deferred until the first forward data pass
- at which time the size will be determined.
+ in_size (int or None): Dimension of input vectors. If ``None``,
+ parameter initialization will be deferred until the first forward
+ data pass at which time the size will be determined.
out_size (int): Dimensionality of output vectors.
Attributes:
