pub fn extract_parameters(params: Vec<PyTensor>) -> PyResult<Vec<Tensor<f32>>>
Helper function to extract parameters from Python objects