oneflow.nn.functional.batch_norm¶
-
oneflow.nn.functional.
batch_norm
(input: oneflow.Tensor, running_mean: Optional[oneflow.Tensor], running_var: Optional[oneflow.Tensor], weight: Optional[oneflow.Tensor] = None, bias: Optional[oneflow.Tensor] = None, training: bool = False, momentum: float = 0.1, eps: float = 1e-05) → oneflow.Tensor¶ Applies Batch Normalization for each channel across a batch of data.
See
BatchNorm1d
,BatchNorm2d
,BatchNorm3d
for details.