R/layers.R
BatchNorm1dFlat.Rd
`nn.BatchNorm1d`, but first flattens leading dimensions
BatchNorm1dFlat( num_features, eps = 1e-05, momentum = 0.1, affine = TRUE, track_running_stats = TRUE )
number of features
epsilon
momentum
affine
track running statistics
None