pub struct SparseTensor {
pub format: SparseFormat,
pub rows: usize,
pub cols: usize,
pub nnz: usize,
pub device: DeviceType,
pub dtype: DType,
}Expand description
Sparse tensor representation with format information
Fields§
§format: SparseFormat§rows: usize§cols: usize§nnz: usize§device: DeviceType§dtype: DTypeImplementations§
Source§impl SparseTensor
impl SparseTensor
pub fn new( format: SparseFormat, rows: usize, cols: usize, nnz: usize, device: DeviceType, dtype: DType, ) -> TorshResult<Self>
pub fn sparsity(&self) -> f64
pub fn memory_size(&self) -> usize
Trait Implementations§
Source§impl Clone for SparseTensor
impl Clone for SparseTensor
Auto Trait Implementations§
impl Freeze for SparseTensor
impl RefUnwindSafe for SparseTensor
impl Send for SparseTensor
impl Sync for SparseTensor
impl Unpin for SparseTensor
impl UnsafeUnpin for SparseTensor
impl UnwindSafe for SparseTensor
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more
impl<ST, DT> CastableFrom<ST, Initialized, Initialized> for DT
impl<ST, DT> CastableFrom<ST, Uninit, Uninit> for DT
Source§impl<T> CloneToUninit for Twhere
T: Clone,
impl<T> CloneToUninit for Twhere
T: Clone,
Source§impl<T> IntoEither for T
impl<T> IntoEither for T
Source§fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
fn into_either(self, into_left: bool) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left is true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read moreSource§fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
fn into_either_with<F>(self, into_left: F) -> Either<Self, Self> ⓘ
Converts
self into a Left variant of Either<Self, Self>
if into_left(&self) returns true.
Converts self into a Right variant of Either<Self, Self>
otherwise. Read more