Update.

[pytorch.git] / pscan.py

diff --git a/pscan.py b/pscan.py
index a1d410c..d009692 100755 (executable)
--- a/pscan.py
+++ b/pscan.py
@@ -18,7 +18,9 @@ def naive_rec(A, X, Y0):
 
 ######################################################################
 
-# A is NxTx1 and X is NxTxD
+# A is NxTx1
+# X is NxTxD
+# Y0 is NxD
 #
 # Returns Y defined with
 #