huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-17 02:58:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

(hotfix) fix the doc test for indexer (#2970)

Browse Source
This commit is contained in:
Congxian Qiu
2025-05-28 12:13:26 +08:00
committed by GitHub
parent 61ddb9535e
commit cac51fe16a
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
candle-core/src/indexer.rs
View File

@ -226,8 +226,8 @@ where
/// assert_eq!(c.to_vec1::<f32>()?, &[1., 4.]); /// assert_eq!(c.to_vec1::<f32>()?, &[1., 4.]);
/// ///
/// let d = a.i((2.., ..))?; /// let d = a.i((2.., ..))?;
/// assert_eq!(c.shape().dims(), &[2]); /// assert_eq!(d.shape().dims(), &[1, 3]);
/// assert_eq!(c.to_vec1::<f32>()?, &[1., 4.]); /// assert_eq!(d.to_vec2::<f32>()?, &[[6., 7., 8.]]);
/// # Ok::<(), candle_core::Error>(()) /// # Ok::<(), candle_core::Error>(())
/// ``` /// ```
fn i(&self, (a, b): (A, B)) -> Result<Tensor, Error> { fn i(&self, (a, b): (A, B)) -> Result<Tensor, Error> {