From 24557d01df2fea576a593bc039318e21a06f7ae4 Mon Sep 17 00:00:00 2001 From: =?utf8?q?Fran=C3=A7ois=20Fleuret?= Date: Mon, 18 Dec 2023 05:13:24 +0100 Subject: [PATCH] Update. --- pscan.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/pscan.py b/pscan.py index 3526c31..c547823 100755 --- a/pscan.py +++ b/pscan.py @@ -114,3 +114,8 @@ if __name__ == "__main__": print((gA - gA_ref).norm()) print((gX - gX_ref).norm()) print((gY_init - gY_init_ref).norm()) + + Y1 = pscan(A[:, : T // 2], X[:, : T // 2], Y_init) + Y2 = pscan(A[:, T // 2 :], X[:, T // 2 :], Y1[:, -1]) + + print((Y - torch.cat([Y1, Y2], dim=1)).norm()) -- 2.20.1