Module auto_diff::op

Re-exports

pub use local::Add;

pub use local::Sub;

pub use local::Mul;

pub use local::Div;

pub use convolution::Conv2d;

pub use linear::Linear;

pub use nonlinear::ELU;

pub use nonlinear::ReLU;

pub use nonlinear::Sigmoid;

pub use loss::MSELoss;

pub use loss::BCEWithLogitsLoss;

pub use loss::CrossEntropyLoss;

Modules

convolution

linear

local

loss

nonlinear

Structs

Nop

Op

Op is the Rc wrapper of typed op trait

View

View op

Traits

OpTrait

All op is OpTrait

Functions

_gradient_checker

Verify the gradient implementation is right.