slop_tensor/sum.rs
1use slop_algebra::AbstractField;
2use slop_alloc::CpuBackend;
3
4use crate::Tensor;
5
6/// Add a scalar value to all elements of a tensor in place.
7pub fn add_assign<T: AbstractField>(lhs: &mut Tensor<T, CpuBackend>, rhs: T) {
8 let lhs = lhs.as_mut_slice();
9 for elem in lhs.iter_mut() {
10 *elem += rhs.clone();
11 }
12}