[−][src]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::PaddingMode; |
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. |