paddlepaddle吧 关注:211贴子:310
  • 1回复贴,共1

请问当conv2d接口激活函数为softmax的时候是不是有问题呢?

只看楼主收藏回复

当激活函数为softmax的时候,输出的值不应该是他们的和为1吗?代码地址
# 分类是否人脸的卷积输出层 conv4_1 = fluid.layers.conv2d(input=conv3_prelu, num_filters=2, filter_size=1, act='softmax', param_attr=ParamAttr(initializer=Xavier(), regularizer=L2DecayRegularizer(0.0005)), name='conv4_1')


1楼2019-06-09 16:02回复
    发现你在conv的输入shape是32,32,1,1;根据softmax的文档,我们目前只支持对最后一维做softmax,而conv得到的shape是32,2,1,1,所以要把conv的输出reshape成32,2,再做softmax


    IP属地:北京2楼2019-06-09 16:02
    回复