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 matmul;
11mod path;
12mod operation;
13mod plan;
14mod tensor;
15
16#[cfg(feature = "cann")]
17pub mod cann;
18
19pub use descriptor::{Mode, OperandDescriptor, TensorDescriptor};
20pub use einsum::{EinsumPlan, einsum, einsum_with_options};
21pub use error::{Error, Result};
22pub use operation::{BinaryOp, ComputeType, OperationDescriptor, ReductionOp, UnaryOp};
23pub use plan::Plan;
24pub use tensor::{contract, elementwise_binary, elementwise_trinary, permute, reduce};
25pub use ruda_core::tensor::DType;
26pub use ruda_kernel::tensor::RudaTensor;