Current section

Files

Jump to
viva_tensor include viva_tensor@nn@cv_BatchNorm2d.hrl
Raw

include/viva_tensor@nn@cv_BatchNorm2d.hrl

-record(batch_norm2d, {
scale :: viva_tensor@tensor:tensor(),
bias :: viva_tensor@tensor:tensor(),
running_mean :: viva_tensor@tensor:tensor(),
running_var :: viva_tensor@tensor:tensor(),
momentum :: float(),
eps :: float()
}).