Allow torch.Tensor inputs to DeepGRU
eonu opened this issue · 1 comments
eonu commented
The _EncoderNetwork
class should be modified to allow a B x T x D
tensor input (when all sequence lengths are the same).
This will involve modifying forward(x, n_lengths)
to make n_lengths
optional.
- If it is set to
None
thenx
should be aB x T x D
tensor. - If
n_lengths
is set, thenx
should be a padded sequence object.
This will also likely require adding conditionals for the following lines: