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§
- DotGeneral
Accumulation - Output-update semantics for dot-general accumulation.
- Eager
Runtime - Shared eager execution context for tensors on a backend.
- Eager
Tensor - Eager tensor with reverse-mode autodiff over concrete tensor values.
- Trace
Context - Mutable owner of one backend-neutral semantic trace.
- Trace
Value - Opaque value handle owned by one
TraceContext. - Traced
Tensor - Typed
Tensor - Runtime typed tensor storage with compile-time scalar type and rank metadata.
- Typed
Tensor View - 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.
- Tensor
Read - Read-only tensor input accepted by synchronous eager kernels.
- Tensor
View - Dynamic read-only borrowed tensor view.
- Tensor
Write - Mutable tensor output accepted by synchronous eager kernels.
- Typed
Tensor Write - Mutable typed tensor output accepted by synchronous eager kernels.
Traits§
- Tensor
Backend - Standard runtime backend over dynamic
Tensorvalues. - Tensor
Scalar - Sealed trait for scalar types that can be stored in a
Tensor.