huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-15 10:26:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: u32 from_be_bytes (#765)

Browse Source
This commit is contained in:
YangNianYi
2023-09-09 15:55:35 +08:00
committed by GitHub
parent c88d6fd4b9
commit 976a1086ee
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
candle-datasets/src/vision/mnist.rs
View File

@ -9,12 +9,9 @@ use std::fs::File;
use std::io::{self, BufReader, Read};
fn read_u32<T: Read>(reader: &mut T) -> Result<u32> {
let mut b = vec![0u8; 4];
let mut b = [0u8; 4];
reader.read_exact(&mut b)?;
let (result, _) = b.iter().rev().fold((0u64, 1u64), |(s, basis), &x| {
(s + basis * u64::from(x), basis * 256)
});
Ok(result as u32)
Ok(u32::from_be_bytes(b))
}
fn check_magic_number<T: Read>(reader: &mut T, expected: u32) -> Result<()> {