Module op

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

comparison

convolution

element

index_slicing

linalg

linear

local

loss

macros

nonlinear

pooling

reduction

vision

Structs

Op

Op is the Rc wrapper of typed op trait

OpHandle

View

View op

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