Crate ad_tensors_rs

Crate ad_tensors_rs 

Source
Expand description

ad-tensors-rs: AD-aware tensor interface layer on top of tenferro-rs.

Re-exports§

pub use ad_value::AdMode;
pub use ad_value::AdScalar;
pub use ad_value::AdTensor;
pub use ad_value::AdValue;
pub use ad_value::NodeId;
pub use ad_value::TapeId;
pub use api::cholesky;
pub use api::cholesky_ad;
pub use api::det;
pub use api::det_ad;
pub use api::eig;
pub use api::eig_ad;
pub use api::eigen;
pub use api::eigen_ad;
pub use api::einsum;
pub use api::einsum_ad;
pub use api::inv;
pub use api::inv_ad;
pub use api::lstsq;
pub use api::lstsq_ad;
pub use api::lu;
pub use api::lu_ad;
pub use api::matrix_exp;
pub use api::matrix_exp_ad;
pub use api::norm;
pub use api::norm_ad;
pub use api::pinv;
pub use api::pinv_ad;
pub use api::qr;
pub use api::qr_ad;
pub use api::slogdet;
pub use api::slogdet_ad;
pub use api::solve;
pub use api::solve_ad;
pub use api::solve_triangular;
pub use api::solve_triangular_ad;
pub use api::svd;
pub use api::svd_ad;
pub use api::AdEigResult;
pub use api::AdEigenResult;
pub use api::AdLstsqResult;
pub use api::AdLuResult;
pub use api::AdQrResult;
pub use api::AdSlogdetResult;
pub use api::AdSvdResult;
pub use context::set_global_context;
pub use context::try_with_global_context;
pub use context::with_global_context;
pub use context::GlobalContextGuard;
pub use dyn_types::DynAdTensor;
pub use dyn_types::DynAdValue;
pub use dyn_types::DynScalar;
pub use dyn_types::DynTensor;
pub use dyn_types::ScalarType;
pub use error::Error;
pub use error::Result;
pub use policy::DiffPolicy;
pub use runtime::set_default_runtime;
pub use runtime::with_default_runtime;
pub use runtime::RuntimeContext;
pub use traits::AdResult;
pub use traits::AllowedPairs;
pub use traits::Differentiable;
pub use traits::FactorizeOptions;
pub use traits::FactorizeResult;
pub use traits::IndexLike;
pub use traits::OpRule;
pub use traits::TensorKernel;

Modules§

ad_value
api
context
dyn_types
error
policy
runtime
traits