Struct TensorDescriptor 

pub struct TensorDescriptor { /* private fields */ }

Shape, element strides, type, offset, and device for one tensor.

Implementations

impl TensorDescriptor

pub fn contiguous(dtype: DataType, shape: Vec<usize>, device: Device) -> Self

Creates a compact C-order tensor on a device.

pub fn try_strided( dtype: DataType, shape: Vec<usize>, strides: Vec<isize>, byte_offset: usize, device: Device, ) -> Result<Self, TensorError>

Creates an explicitly strided tensor. Strides are measured in elements.

pub const fn dtype(&self) -> DataType

Returns the data type.

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

Returns dimensions in logical axis order.

pub fn strides(&self) -> Option<&[isize]>

Returns element strides, or None for compact C order.

pub const fn byte_offset(&self) -> usize

Returns the byte offset from the allocation start to logical element zero.

pub const fn device(&self) -> Device

Returns the storage device.

pub fn element_count(&self) -> Result<usize, TensorError>

Returns the number of logical elements, including one for a scalar.

pub fn is_c_contiguous(&self) -> bool

Returns whether the tensor is compact in C order.

pub fn required_byte_range(&self) -> Result<Range<usize>, TensorError>

Computes the smallest allocation byte range reachable by the layout.

Trait Implementations

impl Clone for TensorDescriptor

fn clone(&self) -> TensorDescriptor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorDescriptor

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

Formats the value using the given formatter.

impl PartialEq for TensorDescriptor

fn eq(&self, other: &TensorDescriptor) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl Eq for TensorDescriptor

impl StructuralPartialEq for TensorDescriptor