burn-tensor 0.18.0

Tensor library with user-friendly APIs and automatic differentiation support
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

#[burn_tensor_testgen::testgen(calibration)]
mod tests {
    use super::*;
    use burn_tensor::{
        Tensor, TensorData,
        quantization::{Calibration, QuantScheme},
    };

    // NOTE: The scheme variant fields are not important for calibration, only the "main" variant (e.g., per-tensor)
    #[test]
    fn min_max_calibration_range_per_tensor() {
        let tensor = TestTensor::<1>::from_floats([-1.8, -1.0, 0.0, 0.5], &Default::default());
        let scheme = QuantScheme::default();

        let range = scheme.compute_range(&tensor, &Calibration::MinMax);

        range
            .min
            .into_data()
            .assert_eq(&TensorData::from([-1.8]), false);
        range
            .max
            .into_data()
            .assert_eq(&TensorData::from([0.5]), false);
    }
}