search

EdGENetworks / attention-networks-for-classification

Hierarchical Attention Networks for Document Classification in PyTorch
603 stars 132 forks source link

unsqueeze to work with pytorch 0.3 #11

Closed ZeweiChu closed 6 years ago

ZeweiChu commented 6 years ago

add unsqueeze in the code to work with pytorch 0.3 as sum method reduces dimensionality in 0.3