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