如何在三维输入(mini-batch 一维输入)上应用批归一化(Batch Normalization Layer)

官方API文档中只提供了2维和4、5维应用批归一化算子,请问我在2维基础上还有额外1维batch构成的3维数据没有直接的算子吗,只能按batch维对那两维一个个使用BatchNorm1d()吗

****************************************************解答*****************************************************

对于3d的数据(N, C, L),如果要在L维度做batch-norm,可以将前两个维度进行合并,reshape到(N*C, L),调用1d,之后对结果进行Reshape。如果要对C,L操作,可以reshape到(N, C*L)然后调用1D。之后对结果再进行reshape。

对于输入size和输出size可以整数的情况,可以先计算kernel_size再使用AvgPool进行等价计算。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值