Update: matrix vector multiplication seems to work if explicitly done on a vector (torch.mv()) but cannot be deduced from e.g. multiplying shapes (2, 2) x (2, 1).
Would be really nice to know if this is expected as in “not implemented” or if this can be done differently.