Struct TensorComprehension 

pub struct TensorComprehension<T: TensorElement> { /* private fields */ }

Builder for tensor comprehensions

Implementations

impl<T: TensorElement + Copy> TensorComprehension<T>

pub fn new() -> Self

Create a new tensor comprehension builder

pub fn device(self, device: DeviceType) -> Self

Set the device for the tensor

pub fn from_iter<I>(self, iter: I) -> Self

where I: IntoIterator<Item = T>,

Add elements from an iterator

pub fn from_iter_with_shape<I>(self, iter: I, shape: Vec<usize>) -> Self

where I: IntoIterator<Item = T>,

Add elements with explicit shape

pub fn build(self) -> Result<Tensor<T>>

Build the tensor

Trait Implementations

impl<T: TensorElement + Copy> Default for TensorComprehension<T>

fn default() -> Self

Returns the “default value” for a type.