rssn 0.2.9

A comprehensive scientific computing library for Rust, aiming for feature parity with NumPy and SymPy.
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
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95

use std::ffi::c_char;

use crate::ffi_apis::common::from_json_string;
use crate::ffi_apis::common::to_json_string;
use crate::symbolic::tensor::Tensor;

/// Performs tensor addition.
/// Takes two JSON strings representing `Tensor` objects as input,
/// and returns a JSON string representing their sum.
#[unsafe(no_mangle)]
pub extern "C" fn rssn_json_tensor_add(
    t1_json: *const c_char,
    t2_json: *const c_char,
) -> *mut c_char {
    let t1: Option<Tensor> = from_json_string(t1_json);

    let t2: Option<Tensor> = from_json_string(t2_json);

    match (t1, t2) {
        | (Some(tensor1), Some(tensor2)) => {
            match tensor1.add(&tensor2) {
                | Ok(result) => to_json_string(&result),
                | Err(_) => std::ptr::null_mut(),
            }
        },
        | _ => std::ptr::null_mut(),
    }
}

/// Performs scalar multiplication on a tensor.
/// Takes a JSON string representing a `Tensor` and a JSON string representing an `Expr` (scalar).
/// Returns a JSON string representing the resulting `Tensor`.
#[unsafe(no_mangle)]
pub extern "C" fn rssn_json_tensor_scalar_mul(
    t_json: *const c_char,
    scalar_json: *const c_char,
) -> *mut c_char {
    let t: Option<Tensor> = from_json_string(t_json);

    let scalar: Option<crate::symbolic::core::Expr> = from_json_string(scalar_json);

    match (t, scalar) {
        | (Some(tensor), Some(s)) => {
            match tensor.scalar_mul(&s) {
                | Ok(result) => to_json_string(&result),
                | Err(_) => std::ptr::null_mut(),
            }
        },
        | _ => std::ptr::null_mut(),
    }
}

/// Computes the outer product of two tensors.
/// Takes two JSON strings representing `Tensor` objects as input,
/// and returns a JSON string representing their outer product.
#[unsafe(no_mangle)]
pub extern "C" fn rssn_json_tensor_outer_product(
    t1_json: *const c_char,
    t2_json: *const c_char,
) -> *mut c_char {
    let t1: Option<Tensor> = from_json_string(t1_json);

    let t2: Option<Tensor> = from_json_string(t2_json);

    match (t1, t2) {
        | (Some(tensor1), Some(tensor2)) => {
            match tensor1.outer_product(&tensor2) {
                | Ok(result) => to_json_string(&result),
                | Err(_) => std::ptr::null_mut(),
            }
        },
        | _ => std::ptr::null_mut(),
    }
}

/// Performs tensor contraction.
/// Takes a JSON string representing a `Tensor`, and two `usize` values representing the axes to contract.
/// Returns a JSON string representing the contracted `Tensor`.
#[unsafe(no_mangle)]
pub extern "C" fn rssn_json_tensor_contract(
    t_json: *const c_char,
    axis1: usize,
    axis2: usize,
) -> *mut c_char {
    let t: Option<Tensor> = from_json_string(t_json);

    if let Some(tensor) = t {
        match tensor.contract(axis1, axis2) {
            | Ok(result) => to_json_string(&result),
            | Err(_) => std::ptr::null_mut(),
        }
    } else {
        std::ptr::null_mut()
    }
}