pytorch BatchNorm2d 函数
此函数主要是为了归一化,本文主要讲解如何计算的,原理可以搜索其他博文。上代码:# -*-coding:utf-8-*-import randomimport numpy as npimport torchtorch.manual_seed(50)random.seed(50)batch = torch.nn.BatchNorm2d(1,momentum = 0, eps=0,affine =False, track_running_stats=False)data =torch.
原创
2022-03-08 11:34:52 ·
1377 阅读 ·
0 评论