Struct Tensor2D 

pub struct Tensor2D { /* private fields */ }

Owned 2D tensor with aligned memory (row-major).

Implementations

impl Tensor2D

pub fn zeros(rows: usize, cols: usize) -> Self

Create a new zero-initialized 2D tensor.

pub fn from_vec(v: Vec<f32>, rows: usize, cols: usize) -> Self

Create from Vec with shape.

pub fn rows(&self) -> usize

pub fn cols(&self) -> usize

pub fn stride(&self) -> usize

pub fn as_ptr(&self) -> *const f32

pub fn as_mut_ptr(&mut self) -> *mut f32

pub fn row(&self, r: usize) -> &[f32]

Get a row slice.

pub fn row_mut(&mut self, r: usize) -> &mut [f32]

Get a mutable row slice.

pub fn row_ptr(&self, r: usize) -> *const f32

Get raw row pointer (includes stride padding).

pub fn row_ptr_mut(&mut self, r: usize) -> *mut f32

Get raw mutable row pointer.

pub fn zero(&mut self)

Fill with zeros.

Trait Implementations

impl Clone for Tensor2D

fn clone(&self) -> Self

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Drop for Tensor2D

fn drop(&mut self)

Executes the destructor for this type.

impl Send for Tensor2D

impl Sync for Tensor2D