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 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() -> 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()

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 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 neg(&self) -> Tensor

pub fn log1pexp(&self) -> Tensor

pub fn sigmoid(&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 sum(&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 !=.