flashlight-tensor

Tensor library written in pure rust, designed mostly for matrix operations

Features

• n-dimensional tensors
• Element-wise operations
• Scalar multiplication and addition
• Tensor multiplication and addition
• Matrix transformation
• Dot product
• CPU only, with GPU support in plans

Instalation

[dependencies]
flashlight_tensor = "0.2.0"

Quick Start

use flashlight_tensor::prelude::*;

fn main(){
    let a: Tensor<f32> = Tensor::fill(1.0, &[2, 2]);
}

Documentation

all tensor operations in tensor category

Tests

Run tests with:
cargo test