Rate this Page
★★★★★

torch.nn.functional.batch_norm#

torch.nn.functional.batch_norm(input,running_mean,running_var,weight=None,bias=None,training=False,momentum=0.1,eps=1e-05)[source]#

Apply Batch Normalization for each channel across a batch of data.

SeeBatchNorm1d,BatchNorm2d,BatchNorm3d for details.

Return type

Tensor