Struct auto_diff::tensor::Tensor

pub struct Tensor { /* fields omitted */ }

Methods

impl Tensor

pub fn new() -> Tensor

pub fn is_empty() -> bool

pub fn size(&self) -> Vec<usize>

pub fn numel(&self) -> usize

pub fn get_scale_f32(&self) -> f32

pub fn get_N(&self) -> Tensor

pub fn get_C(&self) -> Tensor

pub fn get_D(&self) -> Tensor

pub fn get_H(&self) -> Tensor

pub fn get_W(&self) -> Tensor

pub fn numel_tensor(&self) -> Tensor

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

pub fn from_vec_f32(input: &[f32], dim: &[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: &[f64]) -> Tensor

pub fn from_record(&self, row: usize, record: &[f32]) -> Result<(), ()>

pub fn swap(&self, o: Tensor)

pub fn fill(size: &[usize], fill_value: f32) -> Tensor

Returns a tensor of size size filled with fill_value.

pub fn fill_like() -> Tensor

pub fn empty(shape: &[usize]) -> Tensor

pub fn new_ones(dim: &[u32]) -> Tensor

pub fn new_zeros(dim: &[u32]) -> Tensor

pub fn zeros_like(&self) -> Tensor

pub fn ones_like(&self) -> 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(&self, tensors: &[&Tensor], dim: usize) -> Tensor

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(&self, sections: &[usize], dim: usize) -> Vec<Tensor>

pub fn squeeze()

pub fn stack()

pub fn t()

pub fn take()

pub fn transpose()

pub fn unbind()

pub fn permute(&self, dim: &[usize]) -> Tensor

pub fn unsqueeze(&mut self, dim: &[usize]) -> &Tensor

Returns a new tensor with a dimension of size one inserted at the specified position.

The returned tensor shares the same underlying data with this tensor.

pub fn condition()

pub fn to_f64(&mut self)

pub fn to_f32(&mut self)

pub fn abs(&self) -> Tensor

pub fn acos(&self) -> Tensor

pub fn asin(&self) -> Tensor

pub fn atan(&self) -> Tensor

pub fn ceil(&self) -> Tensor

pub fn cos(&self) -> Tensor

pub fn cosh(&self) -> Tensor

pub fn exp(&self) -> Tensor

pub fn expm1(&self) -> Tensor

pub fn floor(&self) -> Tensor

pub fn frac(&self) -> Tensor

pub fn lerp(&self, end: &Tensor, weight: f32) -> Tensor

pub fn lerp_weights(&self, end: &Tensor, weight: &Tensor) -> Tensor

pub fn log(&self) -> Tensor

pub fn log10(&self) -> Tensor

pub fn log1p(&self) -> Tensor

pub fn log1pexp(&self) -> Tensor

pub fn log2(&self) -> Tensor

pub fn neg(&self) -> Tensor

pub fn reciprocal(&self) -> Tensor

pub fn round(&self) -> Tensor

pub fn rsqrt(&self) -> Tensor

pub fn sigmoid(&self) -> Tensor

pub fn sign(&self) -> Tensor

pub fn sin(&self) -> Tensor

pub fn sinh(&self) -> Tensor

pub fn sqrt(&self) -> Tensor

pub fn square(&self) -> Tensor

pub fn tan(&self) -> Tensor

pub fn tanh(&self) -> Tensor

pub fn trunc(&self) -> Tensor

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

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

pub fn mean(&self, dim: usize, keepdim: bool) -> Tensor

pub fn std(&self, dim: usize, keepdim: bool) -> Tensor

pub fn sum(&self) -> Tensor

pub fn var(&self, dim: usize, keepdim: bool) -> Tensor

pub fn normalize(&self, mean: &[f32], std: &[f32]) -> Tensor

pub fn normalize_unit(&self) -> Tensor

Trait Implementations

impl Clone for Tensor

fn clone(&self) -> Self

Returns a copy 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.

impl Display for Tensor

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

Formats the value using the given formatter.

impl Eq for Tensor

impl PartialEq<Tensor> for Tensor

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

This method tests for self and other values to be equal, and is used by ==.

#[must_use]fn ne(&self, other: &Rhs) -> bool

This method tests for !=.