burn-store 0.21.0

Storage and serialization infrastructure for Burn
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

use crate::TensorSnapshot;
use burn_core::module::ParamId;
use burn_tensor::{DType, TensorData};

/// Helper to create a test TensorSnapshot
#[allow(dead_code)]
pub fn create_test_snapshot(
    name: String,
    data: Vec<u8>,
    shape: Vec<usize>,
    dtype: DType,
) -> TensorSnapshot {
    TensorSnapshot::from_data(
        TensorData::from_bytes_vec(data, shape, dtype),
        vec![name],
        vec![],
        ParamId::new(),
    )
}