BatchNorm1dFlat: BatchNorm1dFlat

View source: R/layers.R

BatchNorm1dFlatR Documentation

BatchNorm1dFlat

Description

'nn.BatchNorm1d', but first flattens leading dimensions

Usage

BatchNorm1dFlat(
  num_features,
  eps = 1e-05,
  momentum = 0.1,
  affine = TRUE,
  track_running_stats = TRUE
)

Arguments

num_features

number of features

eps

epsilon

momentum

momentum

affine

affine

track_running_stats

track running statistics

Value

None


fastai documentation built on March 21, 2022, 9:07 a.m.