Output Shape和Param参数解释

查看模型时,可以得到下列图形: 

由于输入层和隐藏层是一块建立的,因此查看模型时输入层和隐藏层只有一层dense(Dense),在该层中输出了两个参数,分别是 Output Shape =(None,256),Param=200960。

Output Shape 第一个参数是Batch size(批尺寸),当批尺寸为None时,表示把数据一次性都放进模型中训练;第二个是输入的shape,输入了256个一维数据。

Param是200960,第一层输入数据(1,784),根据输出的计算公式Output = activation ( input \cdot kernel + bias),可知input\cdot kernel结果是(784,256),bias数据是(1,256)。Param = 784 x 256 + 256 = 200960。

总的来说,每一层的Param可以这么计算:

Param = (上一层神经元数量)x (本层的神经元数量) + (本层的神经元数量)

其中,(上一层神经元数量)x (本层的神经元数量):代表的是 input\cdot kernel 的参数

本层的神经元数量:代表的是bias的参数

照着这种思路,检验一下dense_1(Dense)的参数是否是这样的。 在该层中输出了两个参数,Output Shape =(None,10),Param=2570。

输出是(1,10)的数据,因此Output Shape的第二个参数为10,没问题。

Param=256*10+10=2570,因此Param为2570,没问题。

  • 4
    点赞
  • 41
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值