Struct TensorMut 

pub struct TensorMut<'a> {
    pub data: DevicePtrMut,
    pub dtype: DataType,
    pub shape: &'a [usize],
    pub strides: &'a [i64],
    pub byte_offset: usize,
    pub device: DeviceId,
    /* private fields */
}

Mutable, non-owning view of a tensor on any device.

Fields

data: DevicePtrMut

dtype: DataType

shape: &'a [usize]

strides: &'a [i64]

Strides in elements (may be negative, matching DLPack).

byte_offset: usize

Offset in bytes of the element origin from data (DLPack semantics).

device: DeviceId

Implementations

impl<'a> TensorMut<'a>

pub fn new( data: DevicePtrMut, dtype: DataType, shape: &'a [usize], strides: &'a [i64], device: DeviceId, ) -> Self

Construct a zero-offset mutable view. data must remain valid and exclusively borrowed for 'a.

pub fn with_byte_offset(self, byte_offset: usize) -> Self

Set the DLPack-style byte offset of the element origin.

pub fn validate(&self) -> Result<()>

Check the view’s invariants (rank, dtype, offset alignment).

pub fn is_contiguous(&self) -> bool

Whether the view is contiguous row-major.

pub fn numel(&self) -> usize

Number of elements.

pub fn byte_size(&self) -> usize

Logical element byte size (dense; ignores stride gaps). Uses storage_bytes so sub-byte (int4/uint4) types are counted correctly.

pub fn data_ptr_mut<T>(&mut self) -> *mut T

Typed mutable pointer to the element origin, applying byte_offset. Computed with wrapping arithmetic (no deref) — safe to call.