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

Move the StridedIndex in its own module.

Browse Source
This commit is contained in:
laurent
2023-06-21 07:44:36 +01:00
parent 23db8a7da8
commit 3a5405ca6d
4 changed files with 66 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/tensor.rs
View File

@ -232,8 +232,8 @@ impl Tensor {
Ok(Self(Arc::new(tensor_)))
}
pub(crate) fn strided_index(&self) -> crate::storage::StridedIndex {
crate::storage::StridedIndex::new(self.dims(), self.stride())
pub(crate) fn strided_index(&self) -> crate::StridedIndex {
crate::StridedIndex::new(self.dims(), self.stride())
}
pub fn to_vec1<S: crate::WithDType>(&self) -> Result<Vec<S>> {