Function dfdx::tensor_ops::binary_map::sub

source · [−]

pub fn sub<T: Tensor<Dtype = f32>>(lhs: T, rhs: &T::NoTape) -> T

Expand description

Element wise subtraction.

Example:

let a = Tensor2D::new([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let b = Tensor2D::ones();
let r = sub(a, &b); // or `a - &b`
assert_eq!(r.data(), &[[0.0, 1.0, 2.0], [-2.0, -3.0, -4.0]]);