Function dfdx::tensor_ops::add

source ·
pub fn add<S: Shape, E: Dtype, D, T: Tape<E, D> + Merge<R>, R: Default>(
    lhs: Tensor<S, E, D, T>,
    rhs: Tensor<S, E, D, R>
) -> Tensor<S, E, D, T>where
    D: BinaryKernel<BinaryAddKernelOp, E>,
Expand description

Element wise and scalar addition.

Example:

let a = dev.tensor([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let r = a + dev.ones();
assert_eq!(r.array(), [[2.0, 3.0, 4.0], [0.0, -1.0, -2.0]]);

Adding a scalar:

let a = dev.tensor([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let r = a + 1.0;
assert_eq!(r.array(), [[2.0, 3.0, 4.0], [0.0, -1.0, -2.0]]);