首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >模块“”tensorlayer.layers“”没有属性“flatten”“

模块“”tensorlayer.layers“”没有属性“flatten”“
EN

Stack Overflow用户
提问于 2018-12-23 14:27:35
回答 1查看 341关注 0票数 0

我正在尝试构建一个二值化的神经网络,但错误是“模块'tensorlayer.layers‘没有属性'flatten'”

代码语言:javascript
复制
tf.reset_default_graph()
x = tf.placeholder(tf.float32, [None, 1, 48, 1])
net = tl.layers.InputLayer(x, name='input')
net = tl.layers.BinaryConv2d(net, 32, (5, 5), (1, 1), padding='SAME', name='bcnn1')
net = tl.layers.MaxPool2d(net, (2, 2), (2, 2), padding='SAME', name='pool1')
net = tl.layers.BatchNormLayer(net, act=tl.act.htanh, is_train=True,name='bn1')
net = tl.layers.SignLayer(net)
net = tl.layers.BinaryConv2d(net, 64, (5, 5), (1, 1), padding='SAME', name='bcnn2')
net = tl.layers.MaxPool2d(net, (2, 2), (2, 2), padding='SAME', name='pool2')
net = tl.layers.BatchNormLayer(net, act=tl.act.htanh, is_train=True, name='bn2')
net = tl.layers.flatten(net)
net = tl.layers.DenseLayer(net, n_units=40, act = tf.identity, name='output_layer')
EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2018-12-23 16:47:13

没有flatten属性。您正在寻找FlattenLayer属性:

代码语言:javascript
复制
net = tl.layers.FlattenLayer(net)
票数 1
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/53901674

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档