Enum IntTensor 

pub enum IntTensor {
    U8(Tensor<u8>),
    I32(Tensor<i32>),
    U32(Tensor<u32>),
}

Variants

U8(Tensor<u8>)

I32(Tensor<i32>)

U32(Tensor<u32>)

Implementations

impl IntTensor

pub fn id(&self) -> TensorId

pub fn shape(&self) -> &Shape

pub fn dtype(&self) -> DType

pub fn layout(&self) -> &Layout

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

pub fn dim<D: Dim>(&self, dim: D) -> Result<usize>

pub fn element_count(&self) -> usize

pub fn is_contiguous(&self) -> bool

pub fn rank(&self) -> usize

pub fn flatten_all(&self) -> Result<Self>

pub fn eq(&self, value: i32) -> Result<Tensor<bool>>

pub fn ne(&self, value: i32) -> Result<Tensor<bool>>

pub fn masked_fill(&self, mask: &Tensor<bool>, value: i32) -> Result<Self>

Trait Implementations

impl Clone for IntTensor

fn clone(&self) -> IntTensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl From<&Tensor<i32>> for IntTensor

fn from(value: &Tensor<i32>) -> Self

Converts to this type from the input type.

impl From<&Tensor<u8>> for IntTensor

fn from(value: &Tensor<u8>) -> Self

Converts to this type from the input type.

impl From<&Tensor<u32>> for IntTensor

fn from(value: &Tensor<u32>) -> Self

Converts to this type from the input type.

impl From<Tensor<i32>> for IntTensor

fn from(value: Tensor<i32>) -> Self

Converts to this type from the input type.

impl From<Tensor<u8>> for IntTensor

fn from(value: Tensor<u8>) -> Self

Converts to this type from the input type.

impl From<Tensor<u32>> for IntTensor

fn from(value: Tensor<u32>) -> Self

Converts to this type from the input type.