Struct tensorflow::Tensor

pub struct Tensor<T: TensorType> { /* fields omitted */ }

Holds a multi-dimensional array of elements of a single data type.

The data buffer stores elements in row major order. E.g. if data is treated as a vector of T:

  element 0:   index (0, ..., 0)
  element 1:   index (0, ..., 1)
  ...

Methods

impl<T: TensorType> Tensor<T>

pub fn new(dims: &[u64]) -> Self

Creates a new tensor.

The data is initialized to zeros.

pub fn with_values(self, value: &[T]) -> Result<Self>

Sets (copies) the tensor values to the provided ones.

let a = Tensor::new(&[2, 2]).with_values(&[0_i32, 1, 2, 3]).unwrap();

pub fn dims(&self) -> &[u64]

Returns the tensor's dimensions.

Trait Implementations

impl<T: TensorType> From<T> for Tensor<T>

fn from(value: T) -> Self

Performs the conversion.

impl<'a, T: TensorType> From<&'a [T]> for Tensor<T>

fn from(value: &'a [T]) -> Self

Performs the conversion.

impl<T: Eq + TensorType> Eq for Tensor<T> where     T::InnerType: Eq,

impl<T: TensorType + PartialEq> PartialEq<Tensor<T>> for Tensor<T>

fn eq(&self, other: &Tensor<T>) -> bool

This method tests for self and other values to be equal, and is used by ==.

#[must_use] fn ne(&self, other: &Rhs) -> bool

This method tests for !=.

impl<T: Clone + TensorType> Clone for Tensor<T> where     T::InnerType: Clone,

fn clone(&self) -> Tensor<T>

Returns a copy of the value.

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

Performs copy-assignment from source.

impl<T: Debug + TensorType> Debug for Tensor<T> where     T::InnerType: Debug,

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

Formats the value using the given formatter.

impl<T: TensorType> Display for Tensor<T>

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

Formats the value using the given formatter.

impl<T: TensorType> Deref for Tensor<T>

type Target = [T]

The resulting type after dereferencing.

fn deref(&self) -> &[T]

Dereferences the value.

impl<T: TensorType> DerefMut for Tensor<T>

fn deref_mut(&mut self) -> &mut [T]

Mutably dereferences the value.