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 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::Sigmoid;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§
- OpCall
- Ops that first created, then called needs to follow this behavior.
- OpTrait
- Implement operator by this trait to allow the operator be able to stored in the computation graph.
Functions§
- _gradient_
checker - Verify the gradient implementation is right.
- deserialize_
map - deserialize_
seq - serialize_
box