rwth-i6 / pytorch-to-returnn

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

allow more general reshapes #33

Closed vieting closed 3 years ago

vieting commented 3 years ago

Generalize reshape to allow reshapes of (..., a*b, c,...) -> (..., a, b*c,...) or (..., a, b*c,...) -> (..., a*b, c,...).