From: Francois Fleuret 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] ######################################################################