Struct TensorMeta 

pub struct TensorMeta { /* private fields */ }

Implementations

impl TensorMeta

pub fn scalar(dtype: DType, device: Device) -> Self

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

pub fn quantized_from_shape( shape: Vec<usize>, dtype: DType, device: Device, scale: f64, zero_point: i64, ) -> Result<Self, TensorMetaError>

pub fn quantized_per_channel_from_shape( shape: Vec<usize>, dtype: DType, device: Device, scales: Vec<f64>, zero_points: Vec<i64>, axis: usize, ) -> Result<Self, TensorMetaError>

pub fn from_shape_and_strides( shape: Vec<usize>, strides: Vec<usize>, storage_offset: usize, dtype: DType, device: Device, ) -> Result<Self, TensorMetaError>

pub fn quantized_from_shape_and_strides( shape: Vec<usize>, strides: Vec<usize>, storage_offset: usize, dtype: DType, device: Device, scale: f64, zero_point: i64, ) -> Result<Self, TensorMetaError>

pub fn quantized_per_channel_from_shape_and_strides( shape: Vec<usize>, strides: Vec<usize>, storage_offset: usize, dtype: DType, device: Device, scales: Vec<f64>, zero_points: Vec<i64>, axis: usize, ) -> Result<Self, TensorMetaError>

pub fn with_storage_offset(self, storage_offset: usize) -> Self

pub fn with_dtype(self, dtype: DType) -> Self

pub fn with_quantization(self, quantization: QuantizationParams) -> Self

pub fn validate(&self) -> Result<(), TensorMetaError>

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

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

pub fn storage_offset(&self) -> usize

pub fn dtype(&self) -> DType

pub fn device(&self) -> Device

pub fn quantization(&self) -> Option<&QuantizationParams>

pub fn numel(&self) -> usize

pub fn is_contiguous(&self) -> bool

pub fn storage_index_for( &self, index: &[usize], ) -> Result<usize, TensorMetaError>

pub fn fingerprint64(&self) -> u64

Trait Implementations

impl Clone for TensorMeta

fn clone(&self) -> TensorMeta

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for TensorMeta

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

Formats the value using the given formatter.

impl Eq for TensorMeta

impl PartialEq for TensorMeta

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

Equality operator ==.

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

Inequality operator !=.

impl StructuralPartialEq for TensorMeta