提交 b75f1e89 authored 作者: Frederic's avatar Frederic

move flatten to the utils module.

上级 52b83f7a
......@@ -11,6 +11,7 @@ import numpy
import theano
from theano.configparser import config, AddConfigVar, ConfigParam, StrParam
from theano.gof.utils import flatten
compiledir_format_dict = {"platform": platform.platform(),
"processor": platform.processor(),
......@@ -109,16 +110,6 @@ AddConfigVar('compiledir',
allow_override=False))
def flatten(a):
if isinstance(a, (tuple, list, set)):
l = []
for item in a:
l.extend(flatten(item))
return l
else:
return [a]
def cleanup():
"""
Delete keys in old format from the compiledir.
......
......@@ -332,3 +332,12 @@ def type_guard(type1):
return new_f
return wrap
def flatten(a):
if isinstance(a, (tuple, list, set)):
l = []
for item in a:
l.extend(flatten(item))
return l
else:
return [a]
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论