Error message below:
################################################
input.size(-2) == other_2d.size(-2)INTERNAL ASSERT FAILED at “/opt/conda/conda-bld/pytorch_1619593671510/work/aten/src/ATen/native/BatchLinearAlgebra.cpp”:3111, please report a bug to PyTorch.
##############################################
Apparently this is broken, as I switched to pinv and it worked just fine.