From 84f6dc7e109d822aa6bd5440b7a650c98ab69edb Mon Sep 17 00:00:00 2001
From: Alex Auvolat <alex@adnab.me>
Date: Mon, 21 Dec 2015 10:34:16 +0100
Subject: minifix

---
 ctc.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'ctc.py')

diff --git a/ctc.py b/ctc.py
index cf629c1..a0dab80 100644
--- a/ctc.py
+++ b/ctc.py
@@ -34,7 +34,7 @@ class CTC(Brick):
         l_blk = tensor.zeros((S, B))
         l_blk = tensor.set_subtensor(l_blk[1::2,:],l)
 
-        # dimension of alpha :
+        # dimension of alpha (corresponds to alpha hat in the paper) :
         #   T x B x S
         # dimension of c :
         #   T x B
@@ -44,7 +44,8 @@ class CTC(Brick):
                                         probs[0][tensor.arange(B), l[0]],
                                         tensor.zeros((B, S-2))
                                     ], axis=1)
-        c0 = alpha0.sum(axis=2)
+        c0 = alpha0.sum(axis=1)
+        alpha0 = alpha0 / c0[:,None]
 
         # recursion
         def recursion(p, p_mask, prev_alpha, prev_c):
-- 
cgit v1.2.3