projects
/
pytorch.git
/ blobdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
|
commitdiff
|
tree
raw
|
inline
| side by side
Update.
[pytorch.git]
/
pscan.py
diff --git
a/pscan.py
b/pscan.py
index
c547823
..
891ec55
100755
(executable)
--- a/
pscan.py
+++ b/
pscan.py
@@
-55,8
+55,8
@@
class PScan(torch.autograd.Function):
#
# returns Y of same shape as X, with
#
#
# 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):
@staticmethod
def forward(ctx, A, X, Y_init):