Function dfdx::tensor_ops::mul
source · [−]pub fn mul<Lhs, Rhs>(lhs: Lhs, rhs: Rhs) -> Lhswhere
Lhs: Tensor<Dtype = f32>,
Rhs: Tensor<Dtype = f32, Array = Lhs::Array>,
Lhs::Tape: Merge<Rhs::Tape>,
Expand description
Element wise multiplication.
Example:
let a = tensor([[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);
let b = Tensor2D::ones();
let r = mul(a, b); // or `a * b`
assert_eq!(r.data(), &[[1.0, 2.0, 3.0], [-1.0, -2.0, -3.0]]);