一种神经网络通道数搜索方法和装置,能够实现可微分搜索技术能够进行网络通道数搜索问题,在保证网络性能的同时,减少网络的计算复杂度。该方法包括:确定卷积层的输出通道数N,N为正整数(S301);将卷积层输出的特征张量分割为n个子特征张量,每个子特征张量的通道数为N/n,n为可以被N整除的整数,且n≥2(S302);确定n组加权系数,每组加权系数包括多个加权系数,多个加权系数与n个子特征张量中的多个子特征张量一一对应(S303);确定n组加权系数中每组加权系数中的最大值所对应的子特征张量(S304),以得到n个所述最大值所对应的子特征张量;根据n个最大值所对应的子特征张量重新确定卷积层的输出通道数(S305)。张量重新确定卷积层的输出通道数(S305)。张量重新确定卷积层的输出通道数(S305)。张量重新确定卷积层的输出通道数(S305)。
【技术实现步骤摘要】
【国外来华专利技术】
【技术保护点】
【技术特征摘要】
【国外来华专利技术】
【专利技术属性】
技术研发人员:邱畅啸,杨帆,钟刚,
申请(专利权)人:华为技术有限公司,
类型:发明
国别省市:
还没有人留言评论。发表了对其他浏览者有用的留言会获得科技券。