batch
The batchNorm op used for training
Return
BatchNormResult
Parameters
input
a tensor the last axis of which is the features dimension of size C
scale
the combined scale and shift tensor, with shape (2, C)
The batchNorm op used for training
BatchNormResult
a tensor the last axis of which is the features dimension of size C
the combined scale and shift tensor, with shape (2, C)