Skip to main content

rutensor/
lib.rs

1//! Tensor linear algebra on Ruda device tensors.
2//!
3//! Use [`einsum`] for index expressions or construct an [`OperationDescriptor`]
4//! and reuse a [`Plan`] with different buffers and scalar coefficients.
5
6mod descriptor;
7mod einsum;
8mod error;
9mod kernel;
10mod operation;
11mod plan;
12mod tensor;
13
14pub use descriptor::{Mode, OperandDescriptor, TensorDescriptor};
15pub use einsum::{EinsumPlan, einsum, einsum_with_options};
16pub use error::{Error, Result};
17pub use operation::{BinaryOp, ComputeType, OperationDescriptor, ReductionOp, UnaryOp};
18pub use plan::Plan;
19pub use tensor::{contract, elementwise_binary, elementwise_trinary, permute, reduce};
20pub use ruda_core::tensor::DType;
21pub use ruda_kernel::tensor::RudaTensor;