layers.py 文件源码

python
阅读 30 收藏 0 点赞 0 评论 0

项目:Binary-Neural-Networks 作者: akshaychawla 项目源码 文件源码
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.
评论列表
文章目录


问题


面经


文章

微信
公众号

扫码关注公众号