From: François Fleuret <francois@fleuret.org>
Date: Mon, 18 Dec 2023 04:14:35 +0000 (+0100)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=679c210fdde56edfaf2a1859881139e9d6e81664

Update.
---

diff --git a/pscan.py b/pscan.py
index c547823..891ec55 100755
--- a/pscan.py
+++ b/pscan.py
@@ -55,8 +55,8 @@ class PScan(torch.autograd.Function):
     #
     # returns Y of same shape as X, with
     #
-    # Y[:,t] = A[:,0] * Y_init   + X[:,0] if t == 0
-    #        = A[:,t] * Y[:,t-1] + X[:,t] otherwise
+    # Y[:, t] = A[:, 0] * Y_init   + X[:, 0] if t == 0
+    #        = A[:, t] * Y[:, t-1] + X[:, t] otherwise
 
     @staticmethod
     def forward(ctx, A, X, Y_init):