TensorSafe.Layers.BatchNormalization

This module declares the BatchNormalization layer data type.

Documentation

data BatchNormalization :: Nat -> Nat -> Nat -> Type where

A classic BatchNormalization layer with axis, momentum and epsilon parameters

Constructors

BatchNormalization :: BatchNormalization axis momentum epsilon

Instances

Show (BatchNormalization a b c)

Defined in TensorSafe.Layers.BatchNormalization

Methods

showsPrec :: Int -> BatchNormalization a b c -> ShowS

show :: BatchNormalization a b c -> String

showList :: [BatchNormalization a b c] -> ShowS

(KnownNat axis, KnownNat momentum, KnownNat epsilon) => Layer (BatchNormalization axis momentum epsilon)

Defined in TensorSafe.Layers.BatchNormalization

Methods

layer :: BatchNormalization axis momentum epsilon

compile :: BatchNormalization axis momentum epsilon -> InputShape -> CNetwork