X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=pscan.py;h=d009692a8062423a98b52cb2e4c2e93b35d8562b;hb=ec688e5465818e2606e592b9aaf3ef4afb11cb3f;hp=a1d410cdc70f48c8476e761b790669fefdbfaa36;hpb=190d15d92174901bb61b784b766c9ff7cc4bbb92;p=pytorch.git

diff --git a/pscan.py b/pscan.py
index a1d410c..d009692 100755
--- 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
 #