Let A is m x n tensor and B is m x k tensor.

Then size of normal equation tensor is n x k.

but the size of return tensor of torch.lstsq() is m x k.

Why does it works like this?

1 Like

same question! and how do we get the proper n x k result?