catgrad 0.2.1

a categorical deep learning compiler
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

//! Interpretation of core terms
pub mod types;
pub use types::*;

#[allow(clippy::module_inception)]
pub mod interpreter;
pub use interpreter::*;

pub mod backend;
pub mod parameters;
pub mod tensor_op;

pub use crate::category::core::Shape;
pub use backend::{Backend, BackendError};
pub use parameters::Parameters;

#[cfg(all(test, feature = "ndarray-backend"))]
mod tests;

// Create a tensor
pub fn tensor<B: Backend, T: IntoTagged<B, 1>>(
    backend: &B,
    shape: Shape,
    data: &[T],
) -> Result<Value<B>, BackendError> {
    if shape.size() != data.len() {
        return Err(BackendError::ShapeError);
    }
    let tagged = TaggedTensor::from_slice(backend, data, shape)?;
    Ok(Value::Tensor(tagged))
}