Type Alias coaster::tensor::TensorDesc

pub type TensorDesc = Vec<usize>;

Describes the Descriptor of a SharedTensor.

Aliased Type

struct TensorDesc { /* private fields */ }

Trait Implementations

impl ITensorDesc for TensorDesc

fn rank(&self) -> usize

Returns the rank of the Tensor.

fn size(&self) -> usize

Returns the summed up length of all dimensions of the Tensor.

fn dims(&self) -> &Vec<usize>

Returns the dimensions of the Tensor.

fn dims_i32(&self) -> Vec<i32>

Returns the dimensions of the Tensor as Vec.

fn default_stride(&self) -> Vec<usize>

Returns the default stride for an Rust allocated Tensor.

fn default_stride_i32(&self) -> Vec<i32>

Returns the default stride for a Rust allocated Tensor as i32.