From 546c9d6c72acd590018106a06afc5f1efc1be1a0 Mon Sep 17 00:00:00 2001 From: Francois Fleuret Date: Sun, 7 Jun 2020 15:52:14 +0200 Subject: [PATCH] Minor syntactic change. --- ddpol.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ddpol.py b/ddpol.py index 9a1bbc9..6812fdf 100755 --- a/ddpol.py +++ b/ddpol.py @@ -34,7 +34,7 @@ def fit_alpha(x, y, D, a = 0, b = 1, rho = 1e-12): B = torch.cat((B, y.new_zeros(Q.size(0))), 0) M = torch.cat((M, math.sqrt(rho) * Q.t()), 0) - return torch.lstsq(B, M).solution.view(-1)[:D+1] + return torch.lstsq(B, M).solution[:D+1, 0] ###################################################################### -- 2.20.1