[][src]Module autograph::layer

Modules

builders

Structs

Conv2d

A 2D Convolutional layer

Dense

A linear or fully connected layer with an optional bias

MaxPool2d
Relu

Traits

Forward

Trait for forward pass, implemented by layers
Typically this will call a method or custom Trait method on Variable
A layer like Conv2d will implement Forward, and a model composed of layers will also implement forward.

Layer

Trait for Layers
Custom Models should impl Layer