Function dfdx::nn::npz_fwrite

source · [−]

pub fn npz_fwrite<W: Write + Seek, T: NumpyDtype + NumpyShape + WriteNumbers>(
    w: &mut ZipWriter<W>,
    filename: String,
    data: &T
) -> ZipResult<()>

Expand description

Writes data to a new file in a zip archive named filename.

Example:

let mut zip = ZipWriter::new(...);
let linear: Linear<5, 2> = Default::default();
npz_fwrite(&mut zip, "weight.npy".into(), linear.data());