huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Expose a couple layout methods. (#1816)

Browse Source
This commit is contained in:
Laurent Mazare
2024-03-08 10:52:22 +01:00
committed by GitHub
parent 758366160e
commit 9634583781
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/layout.rs
View File

@ -70,7 +70,7 @@ impl Layout {
self.shape.is_fortran_contiguous(&self.stride) self.shape.is_fortran_contiguous(&self.stride)
} }
pub(crate) fn narrow(&self, dim: usize, start: usize, len: usize) -> Result<Self> { pub fn narrow(&self, dim: usize, start: usize, len: usize) -> Result<Self> {
let dims = self.shape().dims(); let dims = self.shape().dims();
if dim >= dims.len() { if dim >= dims.len() {
Err(Error::DimOutOfRange { Err(Error::DimOutOfRange {
@ -99,7 +99,7 @@ impl Layout {
}) })
} }
pub(crate) fn transpose(&self, dim1: usize, dim2: usize) -> Result<Self> { pub fn transpose(&self, dim1: usize, dim2: usize) -> Result<Self> {
let rank = self.shape.rank(); let rank = self.shape.rank();
if rank <= dim1 || rank <= dim2 { if rank <= dim1 || rank <= dim2 {
Err(Error::UnexpectedNumberOfDims { Err(Error::UnexpectedNumberOfDims {
@ -120,7 +120,7 @@ impl Layout {
}) })
} }
pub(crate) fn permute(&self, idxs: &[usize]) -> Result<Self> { pub fn permute(&self, idxs: &[usize]) -> Result<Self> {
let is_permutation = let is_permutation =
idxs.len() == self.shape.rank() && (0..idxs.len()).all(|i| idxs.contains(&i)); idxs.len() == self.shape.rank() && (0..idxs.len()).all(|i| idxs.contains(&i));
if !is_permutation { if !is_permutation {