Skip to main content

Module prelude

Module prelude 

Source
Expand description

Einsum and tensordot extension traits for concrete, eager, and traced values.

Re-exports§

pub use crate::ContractionTree;
pub use crate::EinsumAxis;
pub use crate::EinsumNotation;
pub use crate::EinsumOptimize;
pub use crate::EinsumSubscripts;
pub use crate::TensorDotAxes;
pub use crate::TensorEinsumExt;
pub use crate::TensorEinsumIntoExt;
pub use crate::TensorReadEinsumExt;
pub use crate::TensorReadEinsumIntoExt;
pub use crate::TensorTensordotExt;
pub use crate::TraceContextEinsumExt;
pub use crate::TracedTensorEinsumExt;
pub use crate::TypedTensorEinsumExt;
pub use crate::TypedTensorEinsumIntoExt;
pub use crate::TypedTensorReadEinsumExt;
pub use crate::TypedTensorReadEinsumIntoExt;
pub use crate::TypedTensorTensordotExt;
pub use crate::EagerEinsumExt;
pub use crate::EagerTensorEinsumExt;

Structs§

DotGeneralAccumulation
Output-update semantics for dot-general accumulation.
EagerRuntime
Shared eager execution context for tensors on a backend.
EagerTensor
Eager tensor with reverse-mode autodiff over concrete tensor values.
TraceContext
Mutable owner of one backend-neutral semantic trace.
TraceValue
Opaque value handle owned by one TraceContext.
TracedTensor
TypedTensor
Runtime typed tensor storage with compile-time scalar type and rank metadata.
TypedTensorView
Read-only borrowed view of typed tensor storage with arbitrary strides.

Enums§

DType
Runtime scalar dtype tag.
Tensor
Dynamic tensor enum over the supported scalar types.
TensorRead
Read-only tensor input accepted by synchronous eager kernels.
TensorView
Dynamic read-only borrowed tensor view.
TensorWrite
Mutable tensor output accepted by synchronous eager kernels.
TypedTensorWrite
Mutable typed tensor output accepted by synchronous eager kernels.

Traits§

TensorBackend
Standard runtime backend over dynamic Tensor values.
TensorScalar
Sealed trait for scalar types that can be stored in a Tensor.