openvino_pytorch_layers
openvino_pytorch_layers copied to clipboard
torch.lstsq specific case
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
.