Struct GPUTensor 

pub struct GPUTensor { /* private fields */ }

GPU tensor wrapper

Implementations

impl GPUTensor

pub fn new(data: Vec<f32>, dims: Vec<usize>, context: Arc<GPUContext>) -> Self

pub fn from_tensor(tensor: &Tensor, context: Arc<GPUContext>) -> Self

pub fn to_tensor(&self) -> Tensor

pub fn to_device(&mut self, device: DeviceType)

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

pub fn device(&self) -> DeviceType

pub fn matmul(&self, other: &GPUTensor) -> GPUTensor

Matrix multiplication on GPU

pub fn add(&self, other: &GPUTensor) -> GPUTensor

Element-wise addition

pub fn mul(&self, other: &GPUTensor) -> GPUTensor

Element-wise multiplication

pub fn scale(&self, scalar: f32) -> GPUTensor

Scalar multiplication

pub fn relu(&self) -> GPUTensor

ReLU activation

pub fn softmax(&self) -> GPUTensor

Softmax activation

pub fn sum(&self) -> f32

Sum reduction

pub fn mean(&self) -> f32

Mean reduction