search

rwth-i6 / pytorch-to-returnn

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

rebase and extend torch.cat #79

Closed vieting closed 2 years ago

vieting commented 2 years ago

Rebase torch.cat to use RETURNN's ConcatLayer, which makes it more flexible. In particular, also other dims than feature can now be concatenated.