Skip to main content

Tensor

openinfer_simulator

Struct Tensor

pub struct Tensor<T> {
    pub data: Vec<T>,
    /* private fields */
}

Expand description

Owned tensor container with shape and stride metadata.

Fields§

§data: Vec<T>

Implementations§

impl<T> Tensor<T>

pub fn from_vec(data: Vec<T>) -> Result<Self>

Build a tensor from a flat data vector.

§Example

let t = Tensor::from_vec(vec![1.0f32, 2.0, 3.0])?;

pub fn from_vec_with_opts(data: Vec<T>, opts: TensorOptions) -> Result<Self>

Build a tensor with explicit options.

§Example

let t = Tensor::from_vec_with_opts(
    vec![1.0f32, 2.0, 3.0, 4.0],
    TensorOptions { shape: Some(vec![2, 2]), ..TensorOptions::default() },
)?;

pub fn from_scalar(value: T) -> Self

Create a scalar tensor from a single value.

§Example

let t = Tensor::from_scalar(3.14f32);

pub fn new(data: Vec<T>) -> Self

Create a tensor, panicking on invalid shape.

pub fn len(&self) -> usize

Return the raw data length.

pub fn shape(&self) -> &[usize]

Return the tensor shape.

pub fn strides(&self) -> &[usize]

Return the tensor strides.

pub fn numel(&self) -> usize

Return the logical element count.

pub fn at(&self, indices: &[usize]) -> &T

Access a value by multidimensional indices.

pub fn view(&self, indices: &[usize]) -> TensorView<T>

Create a view starting at the provided indices.

pub fn to_vec(&self) -> Vec<T>
where T: Clone,

Clone the tensor data into a vector.

Trait Implementations§

impl<T: Clone> Clone for Tensor<T>

fn clone(&self) -> Self

Returns a duplicate of the value. Read more

1.0.0 · Source§

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

Performs copy-assignment from source. Read more

impl<T: Debug> Debug for Tensor<T>

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

Formats the value using the given formatter. Read more

impl<T: TensorElement> Fetchable for Tensor<T>

fn fetch(state: &mut RuntimeState, name: &str) -> Result<Self>

impl From<Tensor<BF16>> for TensorValue

fn from(value: Tensor<BF16>) -> Self

Converts to this type from the input type.

impl From<Tensor<Bitset>> for TensorValue

fn from(value: Tensor<Bitset>) -> Self

Converts to this type from the input type.

impl From<Tensor<F16>> for TensorValue

fn from(value: Tensor<F16>) -> Self

Converts to this type from the input type.

impl From<Tensor<F8>> for TensorValue

fn from(value: Tensor<F8>) -> Self

Converts to this type from the input type.

impl From<Tensor<I1>> for TensorValue

fn from(value: Tensor<I1>) -> Self

Converts to this type from the input type.

impl From<Tensor<I2>> for TensorValue

fn from(value: Tensor<I2>) -> Self

Converts to this type from the input type.

impl From<Tensor<I4>> for TensorValue

fn from(value: Tensor<I4>) -> Self

Converts to this type from the input type.

impl From<Tensor<T1>> for TensorValue

fn from(value: Tensor<T1>) -> Self

Converts to this type from the input type.

impl From<Tensor<T2>> for TensorValue

fn from(value: Tensor<T2>) -> Self

Converts to this type from the input type.

impl From<Tensor<U1>> for TensorValue

fn from(value: Tensor<U1>) -> Self

Converts to this type from the input type.

impl From<Tensor<U2>> for TensorValue

fn from(value: Tensor<U2>) -> Self

Converts to this type from the input type.

impl From<Tensor<U4>> for TensorValue

fn from(value: Tensor<U4>) -> Self

Converts to this type from the input type.

impl From<Tensor<bool>> for TensorValue

fn from(value: Tensor<bool>) -> Self

Converts to this type from the input type.

impl From<Tensor<f32>> for TensorValue

fn from(value: Tensor<f32>) -> Self

Converts to this type from the input type.

impl From<Tensor<f64>> for TensorValue

fn from(value: Tensor<f64>) -> Self

Converts to this type from the input type.

impl From<Tensor<i16>> for TensorValue

fn from(value: Tensor<i16>) -> Self

Converts to this type from the input type.

impl From<Tensor<i32>> for TensorValue

fn from(value: Tensor<i32>) -> Self

Converts to this type from the input type.

impl From<Tensor<i64>> for TensorValue

fn from(value: Tensor<i64>) -> Self

Converts to this type from the input type.

impl From<Tensor<i8>> for TensorValue

fn from(value: Tensor<i8>) -> Self

Converts to this type from the input type.

impl From<Tensor<u16>> for TensorValue

fn from(value: Tensor<u16>) -> Self

Converts to this type from the input type.

impl From<Tensor<u32>> for TensorValue

fn from(value: Tensor<u32>) -> Self

Converts to this type from the input type.

impl From<Tensor<u64>> for TensorValue

fn from(value: Tensor<u64>) -> Self

Converts to this type from the input type.

impl From<Tensor<u8>> for TensorValue

fn from(value: Tensor<u8>) -> Self

Converts to this type from the input type.

impl<T> From<Vec<T>> for Tensor<T>

fn from(value: Vec<T>) -> Self

Converts to this type from the input type.

impl<T, const N: usize> Index<[usize; N]> for Tensor<T>

type Output = TensorView<T>

The returned type after indexing.

fn index(&self, index: [usize; N]) -> &Self::Output

Performs the indexing (container[index]) operation. Read more

impl<T: Send> Send for Tensor<T>

Auto Trait Implementations§

impl<T> !Freeze for Tensor<T>

impl<T> !RefUnwindSafe for Tensor<T>

impl<T> !Sync for Tensor<T>

impl<T> Unpin for Tensor<T>
where T: Unpin,

impl<T> UnwindSafe for Tensor<T>
where T: UnwindSafe + RefUnwindSafe,

Blanket Implementations§

impl<T> Any for T
where T: 'static + ?Sized,

fn type_id(&self) -> TypeId

Gets the TypeId of self. Read more

impl<T> Borrow<T> for T
where T: ?Sized,

fn borrow(&self) -> &T

Immutably borrows from an owned value. Read more

impl<T> BorrowMut<T> for T
where T: ?Sized,

fn borrow_mut(&mut self) -> &mut T

Mutably borrows from an owned value. Read more

impl<T> CloneToUninit for T
where T: Clone,

unsafe fn clone_to_uninit(&self, dest: *mut u8)

🔬This is a nightly-only experimental API. (clone_to_uninit)

Performs copy-assignment from self to dest. Read more

impl<T> From<T> for T

fn from(t: T) -> T

Returns the argument unchanged.

impl<T, U> Into<U> for T
where U: From<T>,

fn into(self) -> U

Calls U::from(self).

That is, this conversion is whatever the implementation of From<T> for U chooses to do.

impl<T> IntoEither for T

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 more

fn into_either_with<F>(self, into_left: F) -> Either<Self, Self>
where F: FnOnce(&Self) -> bool,

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

impl<T> Pointable for T

const ALIGN: usize

The alignment of pointer.

type Init = T

The type for initializers.

unsafe fn init(init: <T as Pointable>::Init) -> usize

Initializes a with the given initializer. Read more

unsafe fn deref<'a>(ptr: usize) -> &'a T

Dereferences the given pointer. Read more

unsafe fn deref_mut<'a>(ptr: usize) -> &'a mut T

Mutably dereferences the given pointer. Read more

unsafe fn drop(ptr: usize)

Drops the object pointed to by the given pointer. Read more

impl<T> ToOwned for T
where T: Clone,

type Owned = T

The resulting type after obtaining ownership.

fn to_owned(&self) -> T

Creates owned data from borrowed data, usually by cloning. Read more

fn clone_into(&self, target: &mut T)

Uses borrowed data to replace owned data, usually by cloning. Read more

impl<T, U> TryFrom<U> for T
where U: Into<T>,

type Error = Infallible

The type returned in the event of a conversion error.

fn try_from(value: U) -> Result<T, <T as TryFrom<U>>::Error>

Performs the conversion.

impl<T, U> TryInto<U> for T
where U: TryFrom<T>,

type Error = <U as TryFrom<T>>::Error

The type returned in the event of a conversion error.

fn try_into(self) -> Result<U, <U as TryFrom<T>>::Error>

Performs the conversion.

impl<V, T> VZip<V> for T
where V: MultiLane<T>,

fn vzip(self) -> V