elftausend / custos-math Goto Github PK

General:

• eq
• gt
• gte
• lt
• lte
• ne

Boolean-specific:

• and
• or
• xor
• not

These can be new traits with the same structure as BaseOps: https://github.com/elftausend/custos-math/blob/55a9976b8f135b5202f3ed2a677a9f309bc9d45c/src/ops/arithmetic.rs

Are there any ops provided for slicing or indexing? Or is this planned for the future?

For example:

let device = CPU::new();
let m = Matrix::<i32>::new(&device, (5, 8));
assert_eq!(m.slice(1..3, 2).dims(), (2, 1));

or maybe:

let slice = Matrix::from_slice_of(&m, 1..3, 2.into());

or even:

let coords = Matrix::slice_coords(1..3, 2.into);
let dest = Matrix::<i32>::new(2, 1);
device.assign_slice(&m, &coords, &mut dest);

Sum and Max are already implemented: https://github.com/elftausend/custos-math/tree/55a9976b8f135b5202f3ed2a677a9f309bc9d45c/src/ops

Other common reduce ops include:

• min
• product
• all
• any

this can just be one of the base arithmetic ops here: https://github.com/elftausend/custos-math/blob/55a9976b8f135b5202f3ed2a677a9f309bc9d45c/src/ops/arithmetic.rs