Struct auto_diff::tensor::Tensor

pub struct Tensor { /* fields omitted */ }

Methods

impl Tensor

pub fn new() -> Tensor

pub fn is_empty() -> bool

pub fn from_vec_f32(input: &Vec<f32>, dim: &Vec<usize>) -> Tensor

Create a tensor from a Vec,

let t1 = Tensor::from_vec_f32(&vec![0., 1., 2., 4.,], &vec![2,2]);

pub fn to_vec_f32(&mut self) -> Vec<f32>

pub fn from_vec_f64(i: &Vec<f64>) -> Tensor

pub fn full() -> Tensor

pub fn full_like() -> Tensor

pub fn empty() -> Tensor

pub fn new_ones(dim: &Vec<u32>) -> Tensor

pub fn new_zeros(dim: &Vec<u32>) -> Tensor

pub fn zeros_like(o: &Tensor) -> Tensor

pub fn ones_like(o: &Tensor) -> Tensor

pub fn range(start: f64, step: f64) -> Tensor

pub fn linespace(start: f64, end: f64, steps: u32) -> Tensor

pub fn logspace(start: f64, end: f64, steps: u32, base: f64) -> Tensor

pub fn eye(n: u32, m: u32) -> Tensor

pub fn cat()

pub fn chunk()

pub fn gather()

pub fn index_select()

pub fn masked_select()

pub fn narrow()

pub fn nonzero()

pub fn reshape()

pub fn split()

pub fn squeeze()

pub fn stack()

pub fn t()

pub fn take()

pub fn transpose()

pub fn unbind()

pub fn unsqueeze()

pub fn condition()

pub fn to_f64(&mut self)

pub fn to_f32(&mut self)

pub fn add(&self, o: &Tensor) -> Tensor

pub fn sub(&self, o: &Tensor) -> Tensor

pub fn mul(&self, o: &Tensor) -> Tensor

pub fn div(&self, o: &Tensor) -> Tensor

pub fn mm(&self, o: &Tensor) -> Tensor

pub fn matmul(&self, o: &Tensor) -> Tensor

Trait Implementations

impl Display for Tensor

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

Formats the value using the given formatter.