Struct AddKernel 

pub struct AddKernel;

Elementwise addition kernel.

Computes output = left + right (binary add).

All tensors must have identical shapes. Addition is performed elementwise in row-major order.

Errors

Returns KernelError::InvalidArguments if inputs.len() != 2.

Returns KernelError::ShapeMismatch if left, right, and output do not all share the same shape.

Examples

let shape = vec![1, 4];
let a = Tensor::from_vec(shape.clone(), vec![1.0, 2.0, 3.0, 4.0]).unwrap();
let b = Tensor::from_vec(shape.clone(), vec![10.0, 20.0, 30.0, 40.0]).unwrap();

let mut out = Tensor::zeros(shape).unwrap();
AddKernel.compute(&[&a, &b], &mut out).unwrap();
assert_eq!(out.data(), &[11.0, 22.0, 33.0, 44.0]);

Trait Implementations

impl Kernel for AddKernel

fn compute( &self, inputs: &[&Tensor], output: &mut Tensor, ) -> Result<(), KernelError>

Computes the kernel output in-place.