TensorSafe.Layers

This module exposes all Layers declared at TensorSafe.Layers.

Documentation

data Conv2D :: Nat -> Nat -> Nat -> Nat -> Nat -> Nat -> Type

A 2D Convolutional layer

Instances

Show (Conv2D a b c d e f)

Defined in TensorSafe.Layers.Conv2D

Methods

showsPrec :: Int -> Conv2D a b c d e f -> ShowS

show :: Conv2D a b c d e f -> String

showList :: [Conv2D a b c d e f] -> ShowS

(KnownNat channels, KnownNat filters, KnownNat kernelRows, KnownNat kernelColumns, KnownNat strideRows, KnownNat strideColumns) => Layer (Conv2D channels filters kernelRows kernelColumns strideRows strideColumns)

Defined in TensorSafe.Layers.Conv2D

Methods

layer :: Conv2D channels filters kernelRows kernelColumns strideRows strideColumns

compile :: Conv2D channels filters kernelRows kernelColumns strideRows strideColumns -> InputShape -> CNetwork

data Dense :: Nat -> Nat -> Type

A classic Dense, or FullyConnected, layer with input and output parameters.

Instances

Show (Dense a b)

Defined in TensorSafe.Layers.Dense

Methods

showsPrec :: Int -> Dense a b -> ShowS

show :: Dense a b -> String

showList :: [Dense a b] -> ShowS

(KnownNat input, KnownNat output) => Layer (Dense input output)

Defined in TensorSafe.Layers.Dense

Methods

layer :: Dense input output

compile :: Dense input output -> InputShape -> CNetwork

data Dropout :: Nat -> Nat -> Type

A Dropout layer with rate and seed arguments

Instances

Show (Dropout a b)

Defined in TensorSafe.Layers.Dropout

Methods

showsPrec :: Int -> Dropout a b -> ShowS

show :: Dropout a b -> String

showList :: [Dropout a b] -> ShowS

(KnownNat rate, KnownNat seed) => Layer (Dropout rate seed)

Defined in TensorSafe.Layers.Dropout

Methods

layer :: Dropout rate seed

compile :: Dropout rate seed -> InputShape -> CNetwork

data Flatten

Flattens the dimensions of the shapes to a list of values with shape D1

Instances

Show Flatten

Defined in TensorSafe.Layers.Flatten

Methods

showsPrec :: Int -> Flatten -> ShowS

show :: Flatten -> String

showList :: [Flatten] -> ShowS

Layer Flatten

Defined in TensorSafe.Layers.Flatten

Methods

layer :: Flatten

compile :: Flatten -> InputShape -> CNetwork

data LSTM :: Nat -> Bool -> Type

A LSTM layer with a number of units and a option to return the original sequences.

Instances

Show (LSTM a b)

Defined in TensorSafe.Layers.LSTM

Methods

showsPrec :: Int -> LSTM a b -> ShowS

show :: LSTM a b -> String

showList :: [LSTM a b] -> ShowS

KnownNat units => Layer (LSTM units b)

Defined in TensorSafe.Layers.LSTM

Methods

layer :: LSTM units b

compile :: LSTM units b -> InputShape -> CNetwork

data MaxPooling :: Nat -> Nat -> Nat -> Nat -> Type

A 2D MaxPooling pooling that works for D2 and D3 shapes

Instances

Show (MaxPooling a b c d)

Defined in TensorSafe.Layers.MaxPooling

Methods

showsPrec :: Int -> MaxPooling a b c d -> ShowS

show :: MaxPooling a b c d -> String

showList :: [MaxPooling a b c d] -> ShowS

(KnownNat kernelRows, KnownNat kernelColumns, KnownNat strideRows, KnownNat strideColumns) => Layer (MaxPooling kernelRows kernelColumns strideRows strideColumns)

Defined in TensorSafe.Layers.MaxPooling

Methods

layer :: MaxPooling kernelRows kernelColumns strideRows strideColumns

compile :: MaxPooling kernelRows kernelColumns strideRows strideColumns -> InputShape -> CNetwork

data Relu

A ReLu activation function

Instances

Show Relu

Defined in TensorSafe.Layers.Relu

Methods

showsPrec :: Int -> Relu -> ShowS

show :: Relu -> String

showList :: [Relu] -> ShowS

Layer Relu

Defined in TensorSafe.Layers.Relu

Methods

layer :: Relu

compile :: Relu -> InputShape -> CNetwork

data Sigmoid

A Sigmoid activation function

Instances

Show Sigmoid

Defined in TensorSafe.Layers.Sigmoid

Methods

showsPrec :: Int -> Sigmoid -> ShowS

show :: Sigmoid -> String

showList :: [Sigmoid] -> ShowS

Layer Sigmoid

Defined in TensorSafe.Layers.Sigmoid

Methods

layer :: Sigmoid

compile :: Sigmoid -> InputShape -> CNetwork