Define reduce() in Python 3

MJGao · Sep 30, 2018 · 448d7b0 · 448d7b0
1 parent 4e6646a
commit 448d7b0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/py2.x/dl/rnn.py b/src/py2.x/dl/rnn.py
@@ -1,12 +1,16 @@
 #!/usr/bin/env python
 # -*- coding: UTF-8 -*-
 
-
 from __future__ import print_function
 import numpy as np
 from cnn import element_wise_op
 from activators import ReluActivator, IdentityActivator
 
+try:
+    reduce         # Python 2
+except NameError:  # Python 3
+    from functools import reduce
+
 
 class RecurrentLayer(object):
     '''
@@ -154,4 +158,4 @@ def test():
     l.forward(x[0])
     l.forward(x[1])
     l.backward(d, ReluActivator())
-    return l
+    return l