Re-exports§
pub use self::error::Error;pub use self::error::ErrorKind;pub use self::error::PredictError;pub use self::nn::Module;pub use self::init::Initialize;pub use self::init::InitializeExt;pub use super::propagate::Propagate;pub use super::shape::ModelShape;pub use super::NdResult;pub use super::Result;pub use super::num::*;pub use super::ops::*;pub use super::predict::*;pub use super::setup::*;pub use super::train::*;
Modules§
Macros§
Constants§
- D_MODEL
- The default model size for any given model
- EPSILON
- The default epsilon value for floating point operations
Traits§
- Affine
- Array
Like - Decrement
- Decrement generally describes an object capable of decrementing itself;
- Default
Like - Entry
- Fill
Like - Increment
- Into
Axis - Inverse
- IsSquare
- Mask
Fill - This trait is used to fill an array with a value based on a mask. The mask is a boolean array of the same shape as the array.
- Matmul
- Matpow
- NdLike
- OfType
- Ones
Like - OrInsert
- Sequence
- A trait for sequential data structures; This trait is implemented for iterators that have a known length.
- Sequence
Iter - Store
- Toggle
- Unsqueeze
- Zeros
Like
Functions§
- concat_
iter - Creates an n-dimensional array from an iterator of n dimensional arrays.
- floor_
div - genspace
- hstack
- inverse
- linarr
- linspace
- rangespace
- creates a matrix from the given shape filled with numerical elements [0, n) spaced evenly by 1
- round_
to - Round the given value to the given number of decimal places.
- stack_
iter - Creates a larger array from an iterator of smaller arrays.
- tril
- Returns the lower triangular portion of a matrix.
- triu
- Returns the upper triangular portion of a matrix.
- vstack