pytorch BatchNorm2d 函数
此函数主要是为了归一化,本文主要讲解如何计算的,原理可以搜索其他博文。
上代码:
# -*-coding:utf-8-*-
import random
import numpy as np
import torch
torch.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 ·
1324 阅读 ·
0 评论