search

rwth-i6 / pytorch-to-returnn

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

add torch.bmm #34

Closed vieting closed 3 years ago

vieting commented 3 years ago

Add batch matix-matrix product torch.bmm using existing matmul.