openvino_pytorch_layers icon indicating copy to clipboard operation
openvino_pytorch_layers copied to clipboard

torch.lstsq specific case

Open dkurt opened this issue 2 years ago • 0 comments

torch.lstsq accepts B and A matrices of shapes MxK and MxN correspondingly. This PR enables least-squares solution for https://github.com/EIDOSlab/torchstain/blob/main/torchstain/normalizers/torch_macenko_normalizer.py where K is always equals 2.

dkurt avatar Feb 07 '22 13:02 dkurt