def __init__(self, input):
self.input = input
if "0.9.0" in theano.__version__:
self.output = T.flatten(self.input, outdim=2) # support theano 0.9.0 api
elif "0.10.0" in theano.__version__:
self.output = T.flatten(self.input, ndim=2) # support theano 0.10.0 api
else:
raise NotImplementedError("this version of theano is not supported") # I can't support all versions; I'm only human.
评论列表
文章目录