candle_einops/
lib.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
mod backend;

pub use candle_einops_macros::einops;

pub use backend::Backend;

/// Specifies the operation used to reduce an axis
#[derive(Copy, Clone, Debug)]
pub enum Operation {
    /// Take the minimum value
    Min,
    /// Take the maximum value
    Max,
    /// Sum all elements
    Sum,
    /// Get the mean value
    Mean,
}