Struct Tensor 

pub struct Tensor<'data> { /* private fields */ }

Tensor, can own, or borrow the underlying tensor

Implementations

impl<'data> Tensor<'data>

pub fn shape(&self) -> &[usize]

The shape of the tensor

use smelte-rs::cpu::f32::Tensor;

let tensor = Tensor::zeros(vec![2, 2]);
assert_eq!(tensor.shape(), vec![2, 2]);

pub fn data(&self) -> &[f32]

A slice to the underlying tensor data

use smelte-rs::cpu::f32::Tensor;

let tensor = Tensor::zeros(vec![2, 2]);
assert_eq!(tensor.data(), vec![0.0; 4]);

pub fn data_mut(&mut self) -> &mut [f32]

A mutable slice to the underlying tensor data

use smelte-rs::cpu::f32::Tensor;

let mut tensor = Tensor::zeros(vec![2, 2]);
tensor.data_mut().iter_mut().for_each(|v| *v += 1.0);
assert_eq!(tensor.data(), vec![1.0; 4]);

pub fn zeros(shape: Vec<usize>) -> Self

Creates a new nulled tensor with given shape

use smelte-rs::cpu::f32::Tensor;

let tensor = Tensor::zeros(vec![2, 2]);

pub fn borrowed( data: &'data [f32], shape: Vec<usize>, ) -> Result<Self, TensorError>

Creates a new borrowed tensor with given shape. Can fail if data doesn’t match the shape

use smelte-rs::cpu::f32::Tensor;

let data = [1.0, 2.0, 3.0, 4.0];
let tensor = Tensor::borrowed(&data, vec![2, 2]).unwrap();

pub fn new<T>(data: T, shape: Vec<usize>) -> Result<Self, TensorError>

where T: Into<Cow<'data, [f32]>>,

Creates a new tensor with given shape. Can fail if data doesn’t match the shape

use smelte-rs::cpu::f32::Tensor;

let data = vec![1.0, 2.0, 3.0, 4.0];
let tensor = Tensor::new(data, vec![2, 2]).unwrap();

Trait Implementations

impl<'data> Clone for Tensor<'data>

fn clone(&self) -> Tensor<'data>

Returns a duplicate of the value.

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

Performs copy-assignment from source.

impl<'a> Debug<Tensor<'a>> for Tensor<'a>

fn data(&self) -> &[f32]

TODO

impl<'a> Tensor for Tensor<'a>

fn shape(&self) -> &[usize]

TODO

fn zeros(shape: Vec<usize>) -> Self

TODO

impl<'a> TensorAdd<Tensor<'a>> for Tensor<'a>

fn add(x: &Self, y: &mut Self) -> Result<(), SmeltError>

TODO

impl<'a> TensorAttention<Tensor<'a>> for Tensor<'a>

fn attention( query: &Linear<F32Tensor<'a>>, key: &Linear<F32Tensor<'a>>, value: &Linear<F32Tensor<'a>>, ctx: &mut BertContext<F32Tensor<'a>>, ) -> Result<(), SmeltError>

TODO

impl<'a> TensorGelu<Tensor<'a>> for Tensor<'a>

fn gelu(x: &mut Tensor<'a>) -> Result<(), SmeltError>

TODO

impl<'a> TensorMatmul<Tensor<'a>> for Tensor<'a>

fn matmul(x: &Self, y: &Self, out: &mut Self) -> Result<(), SmeltError>

TODO

impl<'a> TensorMatmulT<Tensor<'a>> for Tensor<'a>

fn matmul_t(x: &Self, y: &Self, out: &mut Self) -> Result<(), SmeltError>

TODO

impl<'a> TensorMul<Tensor<'a>> for Tensor<'a>

fn mul(x: &Self, y: &mut Self) -> Result<(), SmeltError>

TODO

impl<'a> TensorNormalize<Tensor<'a>> for Tensor<'a>

fn normalize(x: &mut Self, epsilon: f32) -> Result<(), SmeltError>

TODO

impl<'a> TensorSelect<Tensor<'a>> for Tensor<'a>

fn select(x: &[usize], weight: &Self, out: &mut Self) -> Result<(), SmeltError>

TODO

impl<'a> TensorSoftmax<Tensor<'a>> for Tensor<'a>

fn softmax(x: &mut Tensor<'a>) -> Result<(), SmeltError>

TODO

impl<'a> TensorTanh<Tensor<'a>> for Tensor<'a>

fn tanh(x: &mut Tensor<'a>) -> Result<(), SmeltError>

TODO

impl<'a> BertOps<Tensor<'a>> for Tensor<'a>

impl<'a> TensorOps<Tensor<'a>> for Tensor<'a>