tensor-safe-0.1.0.1: Create valid deep neural network architectures

Safe HaskellNone
LanguageHaskell2010

TensorSafe.Layers.BatchNormalization

Description

This module declares the BatchNormalization layer data type.

Synopsis

Documentation

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

A classic BatchNormalization layer with axis, momentum and epsilon parameters

Constructors

BatchNormalization :: BatchNormalization axis momentum epsilon 
Instances
Show (BatchNormalization a b c) Source # 
Instance details

Defined in TensorSafe.Layers.BatchNormalization

(KnownNat axis, KnownNat momentum, KnownNat epsilon) => Layer (BatchNormalization axis momentum epsilon) Source # 
Instance details

Defined in TensorSafe.Layers.BatchNormalization

Methods

layer :: BatchNormalization axis momentum epsilon Source #

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