Struct TensorMut 

pub struct TensorMut<'a, T, const N: usize>
where
    T: DeviceRepr + Copy + 'static,
{
    pub data: DeviceSliceMut<'a, T>,
    pub shape: [i32; N],
    pub stride: [i64; N],
}

Mutable view of a device-resident rank-N tensor.

See TensorRef for the rationale behind the relaxed T bound and the stride / broadcast contract.

Fields

data: DeviceSliceMut<'a, T>

Device-resident element storage.

shape: [i32; N]

Extent along each axis (in elements).

stride: [i64; N]

Element stride along each axis.

Implementations

impl<'a, T, const N: usize> TensorMut<'a, T, N>

where T: DeviceRepr + Copy + 'static,

pub fn numel(&self) -> i64

See TensorRef::numel.

pub fn is_contiguous(&self) -> bool

See TensorRef::is_contiguous.

Trait Implementations

impl<'a, T, const N: usize> Debug for TensorMut<'a, T, N>

where T: Debug + DeviceRepr + Copy + 'static,

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.