pub struct TensorData {
pub data: Vec<f32>,
pub shape: Vec<usize>,
}Expand description
Backend-agnostic tensor data container.
This is the “lingua franca” across the trait boundary between rlox-core (raw buffers/slices) and NN backends. Data is stored as flat f32 in row-major order.
Fields§
§data: Vec<f32>§shape: Vec<usize>Implementations§
Source§impl TensorData
impl TensorData
pub fn new(data: Vec<f32>, shape: Vec<usize>) -> Self
pub fn zeros(shape: Vec<usize>) -> Self
pub fn ones(shape: Vec<usize>) -> Self
pub fn from_f64(data: &[f64], shape: Vec<usize>) -> Self
pub fn numel(&self) -> usize
pub fn ndim(&self) -> usize
pub fn is_empty(&self) -> bool
Trait Implementations§
Source§impl Clone for TensorData
impl Clone for TensorData
Source§fn clone(&self) -> TensorData
fn clone(&self) -> TensorData
Returns a duplicate of the value. Read more
1.0.0 · Source§fn clone_from(&mut self, source: &Self)
fn clone_from(&mut self, source: &Self)
Performs copy-assignment from
source. Read moreSource§impl Debug for TensorData
impl Debug for TensorData
Source§impl PartialEq for TensorData
impl PartialEq for TensorData
impl StructuralPartialEq for TensorData
Auto Trait Implementations§
impl Freeze for TensorData
impl RefUnwindSafe for TensorData
impl Send for TensorData
impl Sync for TensorData
impl Unpin for TensorData
impl UnwindSafe for TensorData
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