Function candle_core::pickle::read_all

source ·
pub fn read_all<P: AsRef<Path>>(path: P) -> Result<Vec<(String, Tensor)>>
Expand description

Read all the tensors from a PyTorch pth file.

§Arguments

  • path - Path to the pth file.