深度神经网络中的维度问题

1、单一样本时的维度问题

在这里插入图片描述
在上图中, l = 5 l=5 l=5
正向传播的过程即为:
z [ 1 ] = w [ 1 ] x + b [ 1 ] z^{[1]}=w^{[1]}x+b^{[1]} z[1]=w[1]x+b[1]
其中, z [ 1 ] z^{[1]} z[1]的维度为(3,1),x即 a [ 0 ] a^{[0]} a[0]的维度为(2,1),由此可知 w [ 1 ] w^{[1]} w[1]的维度为(3,2),则 b [ 1 ] b^{[1]} b[1]的维度为(3,1)。
概括的说, z [ 1 ] z^{[1]} z[1]的维度为( n [ 1 ] n^{[1]} n[1],1),x即 a [ 0 ] a^{[0]} a[0]的维度为( n [ 0 ] n^{[0]} n[0],1), w [ 1 ] w^{[1]} w[1]的维度为( n [ 1 ] n^{[1]} n[1] n [ 0 ] n^{[0]} n[0]),则 b [ 1 ] b^{[1]} b[1]的维度为( n [ 1 ] n^{[1]} n[1],1)。(n为每一层中结点个数)
我们以此推论来验证第二层
z [ 2 ] = w [ 2 ] a [ 1 ] + b [ 2 ] z^{[2]}=w^{[2]}a^{[1]}+b^{[2]} z[2]=w[2]a[1]+b[2]
根据推论有, w [ 2 ] w^{[2]} w[2]的维度为( n [ 2 ] n^{[2]} n[2] n [ 1 ] n^{[1]} n[1]),即(5,3); a [ 1 ] a^{[1]} a[1] z [ 1 ] z^{[1]} z[1]的维度相同,为(3,1),因此, w [ 2 ] ∗ a [ 1 ] w^{[2]}*a^{[1]} w[2]a[1]的维度为(5,1),即为 z [ 2 ] z^{[2]} z[2] b [ 2 ] b^{[2]} b[2]的维度。
综上,总结如下:
对于单个样本在 l l l层的深度神经网络中训练时,
w [ l ] w^{[l]} w[l]的维度为( n [ l ] n^{[l]} n[l] n [ l − 1 ] n^{[l-1]} n[l1]
b [ l ] b^{[l]} b[l]的维度为( n [ l ] n^{[l]} n[l],1)
在反向传播时, d w [ l ] dw^{[l]} dw[l] d b [ l ] db^{[l]} db[l]应与 w [ l ] w^{[l]} w[l] b [ l ] b^{[l]} b[l]相同
d w [ l ] dw^{[l]} dw[l]的维度为( n [ l ] n^{[l]} n[l] n [ l − 1 ] n^{[l-1]} n[l1]
d b [ l ] db^{[l]} db[l]的维度为( n [ l ] n^{[l]} n[l],1)

2、多样本时的维度问题

在这里插入图片描述
对于单一样本时我们知道:
z [ 1 ] = w [ 1 ] x + b [ 1 ] z^{[1]}=w^{[1]}x+b^{[1]} z[1]=w[1]x+b[1]
z [ 1 ] z^{[1]} z[1]的维度为( n [ 1 ] n^{[1]} n[1],1)
w [ 1 ] w^{[1]} w[1]的维度为( n 1 n^{1} n1, n [ 0 ] n^{[0]} n[0]
x x x的维度为( n [ 0 ] n^{[0]} n[0],1)
b [ 1 ] b^{[1]} b[1]的维度为( n [ 1 ] n^{[1]} n[1],1)
当样本量为m时:
z [ 1 ] z^{[1]} z[1]的维度为( n [ 1 ] n^{[1]} n[1],m)
w [ 1 ] w^{[1]} w[1]的维度为( n 1 n^{1} n1, n [ 0 ] n^{[0]} n[0]
x x x A [ 0 ] A^{[0]} A[0]的维度为( n [ 0 ] n^{[0]} n[0],m)
b [ 1 ] b^{[1]} b[1]的维度还是为( n [ 1 ] n^{[1]} n[1],1),但计算时会采用python对其进行扩展,扩展为( n [ 1 ] n^{[1]} n[1],m)
综上,总结如下:
z [ l ] , A [ l ] z^{[l]},A^{[l]} z[l],A[l]的维度为( n [ l ] n^{[l]} n[l],m)
反向传播中相同:
d z [ l ] , d A [ l ] dz^{[l]},dA^{[l]} dz[l],dA[l]的维度为( n [ l ] n^{[l]} n[l],m)

  • 1
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值