Skip to main content

TorchTensor

Trait TorchTensor 

Source
pub trait TorchTensor:
    Debug
    + Send
    + Sync {
    // Required methods
    fn device(&self) -> TorchDevice;
    fn data_ptr(&self) -> u64;
    fn size_bytes(&self) -> usize;
    fn shape(&self) -> Vec<usize>;
    fn stride(&self) -> Vec<usize>;
}

Required Methods§

Source

fn device(&self) -> TorchDevice

Source

fn data_ptr(&self) -> u64

Source

fn size_bytes(&self) -> usize

Source

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

Source

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

Implementors§