From: François Fleuret <francois@fleuret.org>
Date: Sat, 16 Dec 2023 13:57:19 +0000 (-0600)
Subject: Update.
X-Git-Url: https://www.fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pytorch.git;a=commitdiff_plain;h=ec688e5465818e2606e592b9aaf3ef4afb11cb3f

Update.
---

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
 #