From: Francois Fleuret <francois@fleuret.org>
Date: Sun, 7 Jun 2020 13:52:14 +0000 (+0200)
Subject: Minor syntactic change.
X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;h=546c9d6c72acd590018106a06afc5f1efc1be1a0;p=pytorch.git

Minor syntactic change.
---

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]
 
 ######################################################################