tenferro-internal-ops 0.5.0

Graph op vocabulary and AD rule implementations for the tenferro workspace.
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

use std::collections::hash_map::DefaultHasher;
use std::hash::{Hash, Hasher};

#[cfg(feature = "autodiff")]
use crate::input_key::TensorInputKey;

#[test]
fn tensor_input_key_clone_eq_and_hash_are_stable() {
    let lhs = TensorInputKey::User { id: 7 };
    let rhs = lhs.clone();

    let mut lhs_hasher = DefaultHasher::new();
    lhs.hash(&mut lhs_hasher);
    let mut rhs_hasher = DefaultHasher::new();
    rhs.hash(&mut rhs_hasher);

    assert_eq!(lhs, rhs);
    assert_eq!(lhs_hasher.finish(), rhs_hasher.finish());
    assert_eq!(format!("{lhs:?}"), "User { id: 7 }");
}

#[test]
#[cfg(feature = "autodiff")]
fn tensor_input_key_tangent_of_derives_a_unique_key() {
    let key = TensorInputKey::User { id: 3 };
    let tangent = key.tangent_of(42);
    assert_eq!(
        tangent,
        TensorInputKey::Tangent {
            of: Box::new(TensorInputKey::User { id: 3 }),
            pass: 42,
        }
    );
}