Struct Tensor 

pub struct Tensor {
    pub shape: Vec<usize>,
    pub dtype: DataType,
    pub device: Device,
    pub data: TensorData,
    pub strides: Vec<usize>,
}

Tensor structure with real data storage

Fields

shape: Vec<usize>

dtype: DataType

device: Device

data: TensorData

strides: Vec<usize>

Implementations

impl Tensor

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

pub fn zeros(shape: &[usize]) -> Self

pub fn ones(shape: &[usize]) -> Self

pub fn from_data(shape: Vec<usize>, data: TensorData, device: Device) -> Self

pub fn numel(&self) -> usize

pub fn size_bytes(&self) -> usize

pub fn index_select(&self, _dim: usize, _indices: &Tensor) -> Result<Tensor>

pub fn matmul(&self, other: &Tensor) -> Result<Tensor>

pub fn add(&self, other: &Tensor) -> Result<Tensor>

pub fn arange(start: i64, end: i64) -> Self

pub fn layer_norm(&self, _weight: &Tensor) -> Result<Tensor>

pub fn gelu(&self) -> Result<Tensor>

pub fn tanh(&self) -> Result<Tensor>

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Tensor

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl Debug for Tensor

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

Formats the value using the given formatter.