Re-exports
pub use local::Add;
pub use local::Sub;
pub use local::Mul;
pub use local::Div;
pub use local::Matmul;
pub use local::Outer;
pub use linear::Linear;
pub use nonlinear::ELU;
pub use nonlinear::ReLU;
pub use nonlinear::Sigmoid;
pub use convolution::Conv2d;
pub use loss::MSELoss;
pub use loss::BCEWithLogitsLoss;
pub use loss::CrossEntropyLoss;
pub use element::Abs;
pub use element::Acos;
pub use element::Asin;
pub use element::Atan;
pub use element::Ceil;
pub use element::Cos;
pub use element::Cosh;
pub use element::Exp;
pub use element::Expm1;
pub use element::Floor;
pub use element::Frac;
pub use element::Log;
pub use element::Log10;
pub use element::Log1p;
pub use element::Log1pexp;
pub use element::Log2;
pub use element::Neg;
pub use element::Reciprocal;
pub use element::Round;
pub use element::Rsqrt;
pub use element::Sign;
pub use element::Sin;
pub use element::Sinh;
pub use element::Sqrt;
pub use element::Tan;
pub use element::Tanh;
pub use element::Trunc;
pub use comparison::MaxPair;
pub use comparison::MinPair;
pub use comparison::ArgSort;
pub use comparison::EqElem;
pub use comparison::Equal;
pub use comparison::Ge;
pub use comparison::Gt;
pub use comparison::Le;
pub use comparison::Lt;
pub use comparison::Ne;
pub use index_slicing::Cat;
pub use index_slicing::Chunk;
pub use index_slicing::ConditionalSelect;
pub use index_slicing::Gather;
pub use index_slicing::IndexSelect;
pub use index_slicing::IndexExclude;
pub use index_slicing::Reshape;
pub use index_slicing::Split;
pub use index_slicing::Squeeze;
pub use index_slicing::Stack;
pub use index_slicing::T;
pub use index_slicing::Take;
pub use index_slicing::Permute;
pub use index_slicing::Unsqueeze;
pub use index_slicing::Repeat;
pub use linalg::Det;
pub use linalg::Inv;
pub use linalg::NormalizeUnit;
pub use linalg::Tr;
pub use reduction::Argmax;
pub use reduction::Argmin;
pub use reduction::Logsumexp;
pub use reduction::Mean;
pub use reduction::Prod;
pub use reduction::Std;
pub use reduction::Sum;
pub use reduction::Variance;
pub use reduction::Max;
pub use reduction::Min;
pub use vision::GetPatch;
pub use vision::SetPatch;
Modules
Structs
Traits
Ops that first created, then called needs to follow this behavior.
Implement operator by this trait to allow the operator be able to stored in the computation graph.
Functions
Verify the gradient implementation is right.