search

rwth-i6 / pytorch-to-returnn

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

add tensor.expand #10

Closed vieting closed 3 years ago

vieting commented 3 years ago

Add tensor.expand. To do this, first reshape to get additional axes if necessary and then use RETURNN's TileLayer to expand to larger size.