From: François Fleuret Date: Mon, 18 Dec 2023 04:14:35 +0000 (+0100) Subject: Update. X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=commitdiff_plain;ds=inline;h=679c210fdde56edfaf2a1859881139e9d6e81664;hp=24557d01df2fea576a593bc039318e21a06f7ae4;p=pytorch.git 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):