I have a 2D tensor [64,149] were 64 is the batch size and a tensor [64,149,128] (64 is the batch size, too). I would like to get the tensor [64,128]. torch.matmul has not worked so far as it provides me torch.Size([64, 64, 128]) as a result and torch.bmm throws the following error:
*** RuntimeError: invalid argument 6: expected 3D tensor at /opt/conda/conda-bld/pytorch_1570910687230/work/aten/src/THC/generic/THCTensorMathBlas.cu:518