ndrs 0.4.0

A tensor library with GPU support
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

// ndrs/src/lib.rs
pub mod backend;
pub mod device;
pub mod dtype;
pub mod kernel;
pub mod macros;
pub mod stream;
pub mod tensor;
pub mod view;
pub mod view_ops;

pub use backend::cpu;
pub use backend::cuda;

pub use device::Device;
pub use dtype::{
    BinaryOpFn, BinaryOpKind, DTYPE_FLOAT32, DTYPE_INT32, DType, TypeInfo, allocate_dtype,
    get_add_op, get_binary_op, register_binary_op, register_dtype,
};
pub use tensor::Tensor;
pub use view::{ArcTensorView, RcTensorView, SliceInfo, TensorViewOps, broadcast_shapes};

#[cfg(test)]
mod test_init {
    use ctor::ctor;

    #[ctor]
    fn init_logger() {
        let _ = env_logger::builder().is_test(true).try_init();
    }
}