search

rwth-i6 / pytorch-to-returnn

Make PyTorch code runnable within RETURNN
3 stars 6 forks source link

Fix F.pad in case the padding happens in multiple dimensions #44

Closed lucasoethe closed 3 years ago

lucasoethe commented 3 years ago

The current implementation of F.pad fails when the padding tuple has more than 2 entries, because RETURNN expects the padding to be split up by axes.