Doraemonzzz / hgru-pytorch

26 stars 2 forks source link

Hierarchically Gated Recurrent Neural Network

Input shape: (n, b, d), where n is sequence length, b is batch size and d is feature dim.