allow indexing tensor with none - Githubissues

rwth-i6 / pytorch-to-returnn

Make PyTorch code runnable within RETURNN

3 stars 6 forks source link

allow indexing tensor with none #81

Closed vieting closed 2 years ago

vieting commented 2 years ago

x[None] is the same as x.unsqueeze(0).