NN.Architectures.Affine

Definitions

AffineParams : (x : Axis) -> (y : Axis) -> ConsistentWith y [x] => Type -> Type

  A weight matrix and a bias vector

Totality: total
Visibility: public export

affineImpl : {auto {conArg:14043} : ConsistentWith y [x]} -> Num a => AllAlgebra [x] a => TensorMonoid (x .cont) => TensorMonoid (y .cont) => DPair (Tensor [x] a) (const (AffineParams x y a)) -> Tensor [y] a

Totality: total
Visibility: public export

affinePara : Num a => ConsistentWith y [x] => AllAlgebra [x] a => TensorMonoid (x .cont) => TensorMonoid (y .cont) => Tensor [x] a -\-> Tensor [y] a

Totality: total
Visibility: public export

affineModel : {auto {conArg:14281} : Neg a} -> {auto {conArg:14284} : ConsistentWith y [x]} -> AllAlgebra [x] a => {auto {conArg:14305} : TensorMonoid (x .cont)} -> {auto {conArg:14308} : TensorMonoid (y .cont)} -> Algebra (Ext (y .cont)) (Tensor [x] a) => Random (Tensor [y, x] a) => Random (Tensor [y] a) => Const (Tensor [x] a) -\-> Const (Tensor [y] a)

Totality: total
Visibility: public export